Skip to content

Commit

Permalink
Update example to use latest dpctl API.
Browse files Browse the repository at this point in the history
  • Loading branch information
Diptorup Deb committed Apr 5, 2024
1 parent 223df44 commit 9673d9a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion numba_dpex/examples/kernel/matmul.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ def _arange_reshaped(shape, dtype):
X = dpt.asarray(X)
Y = dpt.asarray(Y)
device = X.device.sycl_device
result = dpt.zeros((5, 5), dtype, device=device)
result = dpt.zeros((5, 5), dtype=dtype, device=device)
X_slm = kapi.LocalAccessor(shape=work_group_size, dtype=dtype)
Y_slm = kapi.LocalAccessor(shape=work_group_size, dtype=dtype)

Expand Down

0 comments on commit 9673d9a

Please sign in to comment.