Skip to content

Commit 72622e0

Browse files
Modified test_async_submit to check status right away
Moved host operations done after kernel submission to improve chance of detecting non-complete status of kernel submission.
1 parent c73190c commit 72622e0

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

dpctl/tests/test_sycl_kernel_submit.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,8 @@ def test_async_submit():
156156
assert isinstance(kern2Kernel, dpctl_prog.SyclKernel)
157157
assert isinstance(kern2Kernel, dpctl_prog.SyclKernel)
158158

159-
n = 1024 * 1024
159+
status_complete = dpctl.event_status_type.complete
160+
n = 1024 * 1024 * 4
160161
X = dpt.empty((3, n), dtype="u4", usm_type="device", sycl_queue=q)
161162
first_row = dpctl_mem.as_usm_memory(X[0])
162163
second_row = dpctl_mem.as_usm_memory(X[1])
@@ -191,14 +192,16 @@ def test_async_submit():
191192
None,
192193
[e1, e2],
193194
)
194-
status_complete = dpctl.event_status_type.complete
195+
e3_st = e3.execution_status
196+
e2_st = e2.execution_status
197+
e1_st = e1.execution_status
195198
assert not all(
196199
[
197200
e == status_complete
198201
for e in (
199-
e1.execution_status,
200-
e2.execution_status,
201-
e3.execution_status,
202+
e1_st,
203+
e2_st,
204+
e3_st,
202205
)
203206
]
204207
)

0 commit comments

Comments
 (0)