Skip to content

Commit 9aceb2d

Browse files
Added tests that handling of partitions with zero EU counts works as designed
1 parent f563d87 commit 9aceb2d

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

dpctl/tests/test_sycl_device.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -360,6 +360,13 @@ def check_create_sub_devices_equally(device):
360360
pytest.fail("create_sub_devices failed")
361361

362362

363+
def check_create_sub_devices_equally_zeros(device):
364+
try:
365+
device.create_sub_devices(partition=0)
366+
except TypeError:
367+
pass
368+
369+
363370
def check_create_sub_devices_by_counts(device):
364371
try:
365372
n = device.max_compute_units / 2
@@ -372,6 +379,13 @@ def check_create_sub_devices_by_counts(device):
372379
pytest.fail("create_sub_devices failed")
373380

374381

382+
def check_create_sub_devices_by_counts_zeros(device):
383+
try:
384+
device.create_sub_devices(partition=(0, 1))
385+
except TypeError:
386+
pass
387+
388+
375389
def check_create_sub_devices_by_affinity_not_applicable(device):
376390
try:
377391
device.create_sub_devices(partition="not_applicable")

0 commit comments

Comments
 (0)