test_sycl_program now uses pytest

Author: oleksandr-pavlyk

dpctl/tests/test_sycl_program.py

@@ -18,95 +18,96 @@
 """
 
 import os
-import unittest
+
+import pytest
 
 import dpctl
 import dpctl.program as dpctl_prog
 
-from ._helper import has_gpu
-
-
-@unittest.skipUnless(has_gpu(), "No OpenCL GPU queues available")
-class TestProgramFromOCLSource(unittest.TestCase):
-    def test_create_program_from_source(self):
-        oclSrc = "                                                             \
-        kernel void add(global int* a, global int* b, global int* c) {         \
-            size_t index = get_global_id(0);                                   \
-            c[index] = a[index] + b[index];                                    \
-        }                                                                      \
-        kernel void axpy(global int* a, global int* b, global int* c, int d) { \
-            size_t index = get_global_id(0);                                   \
-            c[index] = a[index] + d*b[index];                                  \
-        }"
-        q = dpctl.SyclQueue("opencl:gpu")
-        prog = dpctl_prog.create_program_from_source(q, oclSrc)
-        self.assertIsNotNone(prog)
-
-        self.assertTrue(prog.has_sycl_kernel("add"))
-        self.assertTrue(prog.has_sycl_kernel("axpy"))
-
-        addKernel = prog.get_sycl_kernel("add")
-        axpyKernel = prog.get_sycl_kernel("axpy")
-
-        self.assertEqual(addKernel.get_function_name(), "add")
-        self.assertEqual(axpyKernel.get_function_name(), "axpy")
-        self.assertEqual(addKernel.get_num_args(), 3)
-        self.assertEqual(axpyKernel.get_num_args(), 4)
-
-
-@unittest.skipUnless(has_gpu(), "No OpenCL GPU queues available")
-class TestProgramFromSPRIV(unittest.TestCase):
-    def test_create_program_from_spirv(self):
-
-        CURR_DIR = os.path.dirname(os.path.abspath(__file__))
-        spirv_file = os.path.join(CURR_DIR, "input_files/multi_kernel.spv")
-        with open(spirv_file, "rb") as fin:
-            spirv = fin.read()
-            q = dpctl.SyclQueue("opencl:gpu")
-            prog = dpctl_prog.create_program_from_spirv(q, spirv)
-            self.assertIsNotNone(prog)
-            self.assertTrue(prog.has_sycl_kernel("add"))
-            self.assertTrue(prog.has_sycl_kernel("axpy"))
-
-            addKernel = prog.get_sycl_kernel("add")
-            axpyKernel = prog.get_sycl_kernel("axpy")
-
-            self.assertEqual(addKernel.get_function_name(), "add")
-            self.assertEqual(axpyKernel.get_function_name(), "axpy")
-            self.assertEqual(addKernel.get_num_args(), 3)
-            self.assertEqual(axpyKernel.get_num_args(), 4)
-
-
-@unittest.skipUnless(
-    has_gpu(backend=dpctl.backend_type.level_zero),
-    "No Level0 GPU queues available",
+
+def get_spirv_abspath(fn):
+    curr_dir = os.path.dirname(os.path.abspath(__file__))
+    spirv_file = os.path.join(curr_dir, "input_files", fn)
+    return spirv_file
+
+
+def test_create_program_from_source_ocl():
+    oclSrc = "                                                             \
+    kernel void add(global int* a, global int* b, global int* c) {         \
+        size_t index = get_global_id(0);                                   \
+        c[index] = a[index] + b[index];                                    \
+    }                                                                      \
+    kernel void axpy(global int* a, global int* b, global int* c, int d) { \
+        size_t index = get_global_id(0);                                   \
+        c[index] = a[index] + d*b[index];                                  \
+    }"
+    try:
+        q = dpctl.SyclQueue("opencl")
+    except dpctl.SyclQueueCreationError:
+        pytest.skip("No OpenCL queue is available")
+    prog = dpctl_prog.create_program_from_source(q, oclSrc)
+    assert prog is not None
+
+    assert prog.has_sycl_kernel("add")
+    assert prog.has_sycl_kernel("axpy")
+
+    addKernel = prog.get_sycl_kernel("add")
+    axpyKernel = prog.get_sycl_kernel("axpy")
+
+    assert "add" == addKernel.get_function_name()
+    assert "axpy" == axpyKernel.get_function_name()
+    assert 3 == addKernel.get_num_args()
+    assert 4 == axpyKernel.get_num_args()
+
+
+def test_create_program_from_spirv_ocl():
+    try:
+        q = dpctl.SyclQueue("opencl")
+    except dpctl.SyclQueueCreationError:
+        pytest.skip("No OpenCL queue is available")
+    spirv_file = get_spirv_abspath("multi_kernel.spv")
+    with open(spirv_file, "rb") as fin:
+        spirv = fin.read()
+    prog = dpctl_prog.create_program_from_spirv(q, spirv)
+    assert prog is not None
+    assert prog.has_sycl_kernel("add")
+    assert prog.has_sycl_kernel("axpy")
+
+    addKernel = prog.get_sycl_kernel("add")
+    axpyKernel = prog.get_sycl_kernel("axpy")
+
+    assert "add" == addKernel.get_function_name()
+    assert "axpy" == axpyKernel.get_function_name()
+    assert 3 == addKernel.get_num_args()
+    assert 4 == axpyKernel.get_num_args()
+
+
+def test_create_program_from_spirv_l0():
+    try:
+        q = dpctl.SyclQueue("level_zero")
+    except dpctl.SyclQueueCreationError:
+        pytest.skip("No Level-zero queue is available")
+    spirv_file = get_spirv_abspath("multi_kernel.spv")
+    with open(spirv_file, "rb") as fin:
+        spirv = fin.read()
+    dpctl_prog.create_program_from_spirv(q, spirv)
+
+
+@pytest.mark.xfail(
+    reason="Level-zero backend does not support compilation from source"
 )
-class TestProgramForLevel0GPU(unittest.TestCase):
-
-    import sys
-
-    def test_create_program_from_spirv(self):
-        CURR_DIR = os.path.dirname(os.path.abspath(__file__))
-        spirv_file = os.path.join(CURR_DIR, "input_files/multi_kernel.spv")
-        with open(spirv_file, "rb") as fin:
-            spirv = fin.read()
-            q = dpctl.SyclQueue("level_zero:gpu")
-            dpctl_prog.create_program_from_spirv(q, spirv)
-
-    @unittest.expectedFailure
-    def test_create_program_from_source(self):
-        oclSrc = "                                                             \
-        kernel void add(global int* a, global int* b, global int* c) {         \
-            size_t index = get_global_id(0);                                   \
-            c[index] = a[index] + b[index];                                    \
-        }                                                                      \
-        kernel void axpy(global int* a, global int* b, global int* c, int d) { \
-            size_t index = get_global_id(0);                                   \
-            c[index] = a[index] + d*b[index];                                  \
-        }"
-        q = dpctl.SyclQueue("level_zero:gpu")
-        dpctl_prog.create_program_from_source(q, oclSrc)
-
-
-if __name__ == "__main__":
-    unittest.main()
+def test_create_program_from_source_l0():
+    try:
+        q = dpctl.SyclQueue("level_zero")
+    except dpctl.SyclQueueCreationError:
+        pytest.skip("No Level-zero queue is available")
+    oclSrc = "                                                             \
+    kernel void add(global int* a, global int* b, global int* c) {         \
+        size_t index = get_global_id(0);                                   \
+        c[index] = a[index] + b[index];                                    \
+    }                                                                      \
+    kernel void axpy(global int* a, global int* b, global int* c, int d) { \
+        size_t index = get_global_id(0);                                   \
+        c[index] = a[index] + d*b[index];                                  \
+    }"
+    dpctl_prog.create_program_from_source(q, oclSrc)