Skip to content

Commit 95e0cf8

Browse files
committed
Eye default dtype now uses _get_dtype
1 parent 8cf376e commit 95e0cf8

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

dpctl/tensor/_ctors.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1102,8 +1102,9 @@ def eye(
11021102
usm_type=usm_type,
11031103
sycl_queue=sycl_queue,
11041104
)
1105-
sycl_queue = normalize_queue_device(sycl_queue=sycl_queue, device=device)
11061105
dpctl.utils.validate_usm_type(usm_type, allow_none=False)
1106+
sycl_queue = normalize_queue_device(sycl_queue=sycl_queue, device=device)
1107+
dtype = _get_dtype(dtype, sycl_queue)
11071108
res = dpt.usm_ndarray(
11081109
(n_rows, n_cols),
11091110
dtype=dtype,

0 commit comments

Comments
 (0)