Cannot serialize arrays

This throws an error:

import numpy as np
import dpnp
import pickle
x = dpnp.array(
    np.sqrt(np.arange(10)).astype(np.float32),
    device="gpu",
)
pickle.dumps(x)

File <stringsource>:2, in dpctl.tensor._usmarray.usm_ndarray.__reduce_cython__()

TypeError: no default __reduce__ due to non-trivial __cinit__

The problem here is that usm_ndarray is not pickleable because dpctl.SyclQueue is not pickleable

In theory, this could maybe, be changed, or at least investigated again

Navigation Menu