shamrock.backends.DeviceBuffer_f32#

class shamrock.backends.DeviceBuffer_f32#

Bases: pybind11_object

Methods

copy_from_stdvec

copy_from_stdvec(self: shamrock.backends.DeviceBuffer_f32, arg0: list[float]) -> None

copy_to_stdvec

copy_to_stdvec(self: shamrock.backends.DeviceBuffer_f32) -> list[float]

fill

fill(self: shamrock.backends.DeviceBuffer_f32, arg0: float) -> None

get_size

get_size(self: shamrock.backends.DeviceBuffer_f32) -> int

get_val_at_idx

get_val_at_idx(self: shamrock.backends.DeviceBuffer_f32, arg0: int) -> float

resize

resize(self: shamrock.backends.DeviceBuffer_f32, arg0: int) -> None

set_val_at_idx

set_val_at_idx(self: shamrock.backends.DeviceBuffer_f32, arg0: int, arg1: float) -> None

reduction performance benchmarks

reduction performance benchmarks
copy_from_stdvec(self: shamrock.backends.DeviceBuffer_f32, arg0: list[float]) None#
copy_to_stdvec(self: shamrock.backends.DeviceBuffer_f32) list[float]#
fill(self: shamrock.backends.DeviceBuffer_f32, arg0: float) None#
get_size(self: shamrock.backends.DeviceBuffer_f32) int#
get_val_at_idx(self: shamrock.backends.DeviceBuffer_f32, arg0: int) float#
resize(self: shamrock.backends.DeviceBuffer_f32, arg0: int) None#
set_val_at_idx(self: shamrock.backends.DeviceBuffer_f32, arg0: int, arg1: float) None#