Skip to content

Commit e14652f

Browse files
committed
Fix local accessor tests on iGPU
1 parent e11ca89 commit e14652f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

numba_dpex/tests/kernel_tests/test_local_accessors.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ def _kernel1(nd_item: NdItem, a, slm):
2626
slm[j] = 0
2727

2828
for m in range(100):
29-
slm[j] += i * m
29+
slm[j] += a.dtype.type(i) * a.dtype.type(m)
3030

3131
a[i] = slm[j]
3232

@@ -40,7 +40,7 @@ def _kernel2(nd_item: NdItem, a, slm):
4040
slm[j] = 0
4141

4242
for m in range(100):
43-
slm[j] += i * m
43+
slm[j] += a.dtype.type(i) * a.dtype.type(m)
4444

4545
a[i] = slm[j]
4646

@@ -58,7 +58,7 @@ def _kernel3(nd_item: NdItem, a, slm):
5858
slm[j] = 0
5959

6060
for m in range(100):
61-
slm[j] += i * m
61+
slm[j] += a.dtype.type(i) * a.dtype.type(m)
6262

6363
a[i] = slm[j]
6464

0 commit comments

Comments
 (0)