Skip to content

Commit e100bb6

Browse files
committed
tests: Tweak omp offloading test to include casts in allocation string
1 parent 5a22a69 commit e100bb6

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

‎tests/test_gpu_openmp.py‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -217,7 +217,8 @@ def test_array_rw(self):
217217
assert len(op.body.allocs) == 2
218218
assert str(op.body.allocs[0]) ==\
219219
('float * r0_vec = (float *)'
220-
'omp_target_alloc(x_size*y_size*z_size*sizeof(float),'
220+
'omp_target_alloc('
221+
'sizeof(float)*(long)z_size*(long)y_size*(long)x_size,'
221222
'omp_get_default_device());')
222223
assert str(op.body.allocs[1]) ==\
223224
'init0(x_M,x_m,y_M,y_m,z_M,z_m,r0_vec,y_size,z_size);'

0 commit comments

Comments
 (0)