Closed
Description
dpnp_array must have __dlpack__(self, stream=None)
, __dlpack_device__(self)
methods and provide dpnp.from_dlpack(obj)
function.
def __dlpack__(self, stream=None):
return self._array_obj.__dlpack__(stream=stream)
def __dlpack_device__(self):
return self._array_obj.__dlpack_device__()
Also:
def from_dlpack(obj):
"Docstring"
usm_ary = dpt.from_dlpack(obj)
dpnp_ary = dpnp_array.__new__()
dpnp_ary._array_obj = usm_ary
return dpnp_ary
It should be possible to use mpi4py
from Intel channel to broadcast dpnp arrays.
Metadata
Metadata
Assignees
Labels
No labels