Skip to content

Commit 5823125

Browse files
author
Diptorup Deb
committed
Fix test cases.
1 parent fa4f29a commit 5823125

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

dpctl/tests/test_sycl_queue_manager.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -86,20 +86,23 @@ def test_get_current_device_type_outside_device_ctxt (self):
8686
def test_get_current_device_type_inside_device_ctxt (self):
8787
self.assertEqual(dpctl.get_current_device_type(), None)
8888

89-
with dpctl.device_context(dpctl.device_type.gpu):
89+
with dpctl.device_context("opencl:gpu:0"):
9090
self.assertEqual(dpctl.get_current_device_type(), dpctl.device_type.gpu)
9191

9292
self.assertEqual(dpctl.get_current_device_type(), None)
9393

94-
@unittest.skipIf(not dpctl.has_cpu_queues(), "No CPU platforms available")
94+
@unittest.skipUnless(dpctl.get_num_queues(backend_ty="opencl",
95+
device_ty="cpu") > 0,
96+
"No OpenCL CPU queues available")
9597
def test_get_current_device_type_inside_nested_device_ctxt (self):
9698
self.assertEqual(dpctl.get_current_device_type(), None)
9799

98-
with dpctl.device_context(dpctl.device_type.cpu):
100+
with dpctl.device_context("opencl:cpu:0"):
99101
self.assertEqual(dpctl.get_current_device_type(), dpctl.device_type.cpu)
100102

101-
with dpctl.device_context(dpctl.device_type.gpu):
102-
self.assertEqual(dpctl.get_current_device_type(), dpctl.device_type.gpu)
103+
with dpctl.device_context("opencl:gpu:0"):
104+
self.assertEqual(dpctl.get_current_device_type(), dpctl.
105+
device_type.gpu)
103106
self.assertEqual(dpctl.get_current_device_type(), dpctl.device_type.cpu)
104107

105108
self.assertEqual(dpctl.get_current_device_type(), None)

0 commit comments

Comments
 (0)