dasf.utils.decorators

Implementations of important library decorators.

Functions

is_forced_local(cls)	Returns if object is forced to run in a CPU.
is_forced_gpu(cls)	Returns if object is forced to run in a GPU.
fetch_from_dask(*args, **kwargs)	Fetches to CPU all parameters in a Dask data type.
fetch_from_gpu(*args, **kwargs)	Fetches to CPU all parameters in a GPU data type.
fetch_args_from_dask(func)	Fetches to CPU all function parameters in a Dask data type.
fetch_args_from_gpu(func)	Fetches to CPU all function parameters in a GPU data type.
task_handler(func)	Returns all mapped functions corresponding to the executor in place.

Module Contents

dasf.utils.decorators.is_forced_local(cls)

Returns if object is forced to run in a CPU.

dasf.utils.decorators.is_forced_gpu(cls)

Returns if object is forced to run in a GPU.

dasf.utils.decorators.fetch_from_dask(*args, **kwargs)

Fetches to CPU all parameters in a Dask data type.

Return type:

tuple

dasf.utils.decorators.fetch_from_gpu(*args, **kwargs)

Fetches to CPU all parameters in a GPU data type.

Return type:

tuple

dasf.utils.decorators.fetch_args_from_dask(func)

Fetches to CPU all function parameters in a Dask data type.

dasf.utils.decorators.fetch_args_from_gpu(func)

Fetches to CPU all function parameters in a GPU data type.

dasf.utils.decorators.task_handler(func)

Returns all mapped functions corresponding to the executor in place.