dpnp.partition segfaults on xpu

[fcharras]

Reproducer:

import numpy as np
import dpnp
import dpctl.tensor as dpt

rng = np.random.default_rng(123)

n = 2 ** 29
k = 100
dtype = np.float32

X = rng.uniform(size=n).astype(dtype)
X = dpt.asarray(X, order="C", dtype=dtype)
dpnp.partition(X, 100)  # segfaults

Tested both on Flex 170 GPU and iris iGPU.

[antonwolfy]

@fcharras , thank you for reporting the issue. I will start looking it.