Some question about cuMemImportFromShareableHandle

[MARD1NO]

Hi cuda-python developers, I meet a problem that when I pass a CUmemFabricHandle to cuMemImportFromShareableHandle, it raise a error like:

Provided argument is of type <class 'cuda.bindings.driver.CUmemFabricHandle'> but expected Type <class 'NoneType'>, <class 'type'> or object with Buffer Protocol

My code is:

import cuda.bindings.driver as cudriver 

# equal to CU_MEM_HANDLE_TYPE_FABRIC
handle_type = cudriver.CUmemAllocationHandleType(8)
peer_handle = cudriver.CUmemFabricHandle()
peer_handle.data = xxx

err, buf = cudriver.cuMemImportFromShareableHandle(peer_handle, handle_type)

Is there something wrong with my code or it just a type check bug in cuda-python?

[MARD1NO]

Or should I pass peer_handle.data or peer_handle.getPtr() ?....

[leofang]

@vzhurba01 could you take a look?

[MARD1NO]

@vzhurba01 could you take a look?

I find it need to use peer_handle.data, thanks :D