Skip to content

Commit 4ab4cb0

Browse files
committed
Add tests for dpctl.SyclPlatform.get_composite_devices
1 parent 1c49cac commit 4ab4cb0

File tree

2 files changed

+19
-1
lines changed

2 files changed

+19
-1
lines changed

dpctl/tests/test_sycl_device.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -335,6 +335,7 @@ def test_get_component_devices_from_composite():
335335
devices = dpctl.get_devices()
336336
composite_devices = dpctl.get_composite_devices()
337337
for Cd in composite_devices:
338-
component_devices = Cd.component_devices
338+
component_devices = Cd.component_devices()
339339
for d in component_devices:
340+
# component devices are root devices
340341
assert d in devices

dpctl/tests/test_sycl_platform.py

+17
Original file line numberDiff line numberDiff line change
@@ -259,3 +259,20 @@ def test_platform_get_devices_enum_device_type():
259259
devices = p.get_devices(device_type=dty)
260260
if len(devices):
261261
assert (d.device_type == dty for d in devices)
262+
263+
264+
def test_platform_get_composite_devices():
265+
platforms = dpctl.get_platforms()
266+
if platforms:
267+
for p in platforms:
268+
composite_devices = p.get_composite_devices()
269+
if not composite_devices:
270+
pytest.skip("No composite devices available")
271+
devices = p.get_devices()
272+
for Cd in composite_devices:
273+
assert Cd.has_aspect_is_composite
274+
component_devices = Cd.component_devices()
275+
# all component devices are root devices
276+
assert all(d in devices for d in component_devices)
277+
else:
278+
pytest.skip("No platforms available")

0 commit comments

Comments
 (0)