Skip to content

Commit aec7ba3

Browse files
author
Diptorup Deb
committed
Fix test cases.
1 parent 96de2e1 commit aec7ba3

File tree

2 files changed

+14
-6
lines changed

2 files changed

+14
-6
lines changed

dpctl/tests/test_sycl_program.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,7 @@ def test_create_program_from_spirv(self):
8282
self.assertEqual(axpyKernel.get_num_args(), 4)
8383

8484
@unittest.skipUnless(
85-
dpctl.has_gpu_queues(backend_ty=dpctl.backend_type.opencl),
85+
dpctl.has_gpu_queues(backend_ty=dpctl.backend_type.level_zero),
8686
"No Level0 GPU queues available"
8787
)
8888
class TestProgramForLevel0GPU (unittest.TestCase):

dpctl/tests/test_sycl_queue.py

Lines changed: 13 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -25,17 +25,25 @@
2525
import dpctl
2626
import unittest
2727

28-
@unittest.skipUnless(dpctl.has_gpu_queues(), "No OpenCL GPU queues available")
29-
@unittest.skipUnless(dpctl.has_cpu_queues(), "No OpenCL GPU queues available")
3028
class TestSyclQueue (unittest.TestCase):
29+
@unittest.skipUnless(
30+
dpctl.has_gpu_queues(), "No OpenCL GPU queues available"
31+
)
32+
@unittest.skipUnless(
33+
dpctl.has_cpu_queues(), "No OpenCL CPU queues available"
34+
)
35+
def test_queue_not_equals (self):
36+
with dpctl.device_context("opencl:gpu") as gpuQ0:
37+
with dpctl.device_context("opencl:cpu") as cpuQ:
38+
self.assertFalse(cpuQ.equals(gpuQ0))
3139

40+
@unittest.skipUnless(
41+
dpctl.has_gpu_queues(), "No OpenCL GPU queues available"
42+
)
3243
def test_queue_equals (self):
3344
with dpctl.device_context("opencl:gpu") as gpuQ0:
3445
with dpctl.device_context("opencl:gpu") as gpuQ1:
3546
self.assertTrue(gpuQ0.equals(gpuQ1))
36-
with dpctl.device_context("opencl:cpu") as cpuQ:
37-
self.assertFalse(cpuQ.equals(gpuQ0))
38-
3947

4048
if __name__ == '__main__':
4149
unittest.main()

0 commit comments

Comments
 (0)