shamrock.algs#

algorithmic library

Functions

benchmark_is_all_true

benchmark_is_all_true(arg0: sham::DeviceBuffer<unsigned char, (sham::USMKindTarget)0>, arg1: int) -> float

benchmark_reduction_sum

benchmark_reduction_sum(*args, **kwargs) Overloaded function.

benchmark_scan_exclusive_sum_in_place

benchmark_scan_exclusive_sum_in_place(arg0: sham::DeviceBuffer<unsigned int, (sham::USMKindTarget)0>, arg1: int) -> float

gen_seed

gen_seed(arg0: int) -> shamrock.algs.rng

get_current_impl_is_all_true

get_current_impl_is_all_true() -> shamrock.algs.impl_param

get_current_impl_reduction

get_current_impl_reduction() -> shamrock.algs.impl_param

get_current_impl_scan_exclusive_sum_in_place

get_current_impl_scan_exclusive_sum_in_place() -> shamrock.algs.impl_param

get_default_impl_list_is_all_true

get_default_impl_list_is_all_true() -> list[shamrock.algs.impl_param]

get_default_impl_list_reduction

get_default_impl_list_reduction() -> list[shamrock.algs.impl_param]

get_default_impl_list_scan_exclusive_sum_in_place

get_default_impl_list_scan_exclusive_sum_in_place() -> list[shamrock.algs.impl_param]

is_all_true

is_all_true(arg0: sham::DeviceBuffer<unsigned char, (sham::USMKindTarget)0>, arg1: int) -> bool

mock_buffer_f64

mock_buffer_f64(arg0: int, arg1: int, arg2: float, arg3: float) -> sham::DeviceBuffer<double, (sham::USMKindTarget)0>

mock_buffer_f64_2

mock_buffer_f64_2(arg0: int, arg1: int, arg2: hipsycl::sycl::vec<double, 2, hipsycl::sycl::detail::vec_storage<double, 2> >, arg3: hipsycl::sycl::vec<double, 2, hipsycl::sycl::detail::vec_storage<double, 2> >) -> sham::DeviceBuffer<hipsycl::sycl::vec<double, 2, hipsycl::sycl::detail::vec_storage<double, 2> >, (sham::USMKindTarget)0>

mock_buffer_f64_3

mock_buffer_f64_3(arg0: int, arg1: int, arg2: f64_3, arg3: f64_3) -> sham::DeviceBuffer<hipsycl::sycl::vec<double, 3, hipsycl::sycl::detail::vec_storage<double, 3> >, (sham::USMKindTarget)0>

mock_buffer_u8

mock_buffer_u8(arg0: int, arg1: int, arg2: int, arg3: int) -> sham::DeviceBuffer<unsigned char, (sham::USMKindTarget)0>

mock_gaussian

mock_gaussian(arg0: shamrock.algs.rng) -> float

mock_gaussian_f64_2

mock_gaussian_f64_2(arg0: shamrock.algs.rng) -> hipsycl::sycl::vec<double, 2, hipsycl::sycl::detail::vec_storage<double, 2> >

mock_gaussian_f64_3

mock_gaussian_f64_3(arg0: shamrock.algs.rng) -> f64_3

mock_unit_vector_f64_3

mock_unit_vector_f64_3(arg0: shamrock.algs.rng) -> f64_3

scan_exclusive_sum_in_place

scan_exclusive_sum_in_place(arg0: sham::DeviceBuffer<unsigned int, (sham::USMKindTarget)0>, arg1: int) -> None

set_impl_is_all_true

set_impl_is_all_true(arg0: str, arg1: str) -> None

set_impl_reduction

set_impl_reduction(arg0: str, arg1: str) -> None

set_impl_scan_exclusive_sum_in_place

set_impl_scan_exclusive_sum_in_place(arg0: str, arg1: str) -> None

sum

sum(arg0: sham::DeviceBuffer<double, (sham::USMKindTarget)0>, arg1: int, arg2: int) -> float

Classes