nums.core.compute.compute_manager module

class nums.core.compute.compute_manager.ComputeManager(system, compute_module, device_grid)[source]

Bases: nums.core.compute.compute_interface.ComputeInterface

Abstraction to support multiple systems; namely simultaneous support for CPU and GPU system implementations.

call(name, *args, **kwargs)[source]
call_actor_method(actor, method, *args, **kwargs)[source]
compute_block_shape(shape, dtype, cluster_shape=None, num_cores=None)[source]
static compute_block_shape_static(shape, dtype, cluster_shape, num_cores)[source]
classmethod create(system, compute_module, device_grid)[source]
classmethod destroy()[source]
devices()[source]
get(object_ids)[source]
get_block_shape(shape, dtype)[source]
get_callable(name)[source]
get_rng(seed)[source]
Return type

RNGInterface

init_compute(compute_module)[source]
instance = None
make_actor(name, *args, device_id=None, **kwargs)[source]
num_cores_total()[source]
put(value, **kwargs)[source]
register(name, func, remote_params=None)[source]
register_actor(name, cls)[source]
remote(function, remote_params)[source]
update_block_shape_map(shape_dim, block_shape_dim)[source]
nums.core.compute.compute_manager.instance()[source]
Return type

ComputeManager