xp.eye(..., device=device) has not effect when xp is cupy #337
Description
Dear all,
I run the following command when xp is cupy
# cupy x = cupy.asarray([1.0] * 3) xp = array_api_compat.array_namespace(x) E = xp.eye(3, 3, k=0, dtype='float64', device=1) print(E.device) with xp.cuda.Device(1): E = xp.eye(3, 3, k=0, dtype='float64', device=1) print(E.device) # torch x = torch.as_tensor([1.0] * 3) xp = array_api_compat.array_namespace(x) E = xp.eye(3, 3, k=0, dtype=torch.float64, device='cuda:1') print(E.device)and I get the result:
<CUDA Device 0> <CUDA Device 1> cuda:1It seems that device argument of xp.eye has no effect here when xp is cupy.
It seems to work well when xp is torch.
Did I miss something?
Thank you.
array-api-compat 1.12.0
cupy-cuda12x 13.4.1