numba_dpex.core.utils.caching_utils module

numba_dpex.core.utils.caching_utils.build_key(*args)

Constructs key from variable list of args

Args:

*args: List of components to construct key

Return:

Tuple of args

numba_dpex.core.utils.caching_utils.create_func_hash(pyfunc)

Creates a tuple of sha256 hashes out of code and variable bytes extracted from the compiled funtion.

Args:

pyfunc: Python function object

Return:

Tuple of hashes of code and variable bytes

numba_dpex.core.utils.caching_utils.strip_usm_metadata(argtypes)

Convert the USMNdArray to an abridged type that disregards the usm_type, device, queue, address space attributes.

Args:

argtypes: List of types

Return:

Tuple of types after removing USM metadata from USMNdArray type