cuquantum.cutensornet.set_device_mem_handler¶

cuquantum.cutensornet.set_device_mem_handler(intptr_t handle, handler)[source]¶

Set the device memory handler for cuTensorNet.

The handler object can be passed in multiple ways:

If handler is an int, it refers to the address of a fully initialized cutensornetDeviceMemHandler_t struct.

If handler is a Python sequence:

If handler is a sequence of length 4, it is interpreted as (ctx, device_alloc, device_free, name), where the first three elements are the pointer addresses (int) of the corresponding members. name is a str as the name of the handler.

If handler is a sequence of length 3, it is interpreted as (malloc, free, name), where the first two objects are Python callables with the following calling convention:

ptr = malloc(size, stream)

free(ptr, size, stream)

with all arguments and return value (ptr) being Python int. name is the same as above.

Note

Only when handler is a length-3 sequence will the GIL be held whenever a routine requires memory allocation and deallocation, so for all other cases be sure your handler does not manipulate any Python objects.

Parameters

handle (intptr_t) – The library handle.
handler – The memory handler object, see above.