-
Notifications
You must be signed in to change notification settings - Fork 23
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