dpnp_array must have __dlpack__(self, stream=None), __dlpack_device__(self) methods and provide dpnp.from_dlpack(obj) function.
def__dlpack__(self, stream=None):
returnself._array_obj.__dlpack__(stream=stream)
def__dlpack_device__(self):
returnself._array_obj.__dlpack_device__()
Also:
deffrom_dlpack(obj):
"Docstring"usm_ary=dpt.from_dlpack(obj)
dpnp_ary=dpnp_array.__new__()
dpnp_ary._array_obj=usm_aryreturndpnp_ary
It should be possible to use mpi4py from Intel channel to broadcast dpnp arrays.
dpnp_array must have
__dlpack__(self, stream=None),__dlpack_device__(self)methods and providedpnp.from_dlpack(obj)function.Also:
It should be possible to use
mpi4pyfrom Intel channel to broadcast dpnp arrays.