Allow program creation only for OpenCL for the time being.

Author: Diptorup Deb

dpctl/sycl_core.pyx

@@ -710,8 +710,14 @@ def create_program_from_source (SyclQueue q, unicode source, unicode copts=""):
                                     syc::program returned by the C API.
     '''
 
-    cdef DPPLSyclProgramRef Pref
+    BE = q.get_sycl_backend()
+    if BE != backend_type.opencl:
+        raise ValueError(
+            "Cannot create program for a ", BE, "type backend. Currently only "
+            "OpenCL devices are supported for program creations."
+        )
 
+    cdef DPPLSyclProgramRef Pref
     cdef bytes bSrc = source.encode('utf8')
     cdef bytes bCOpts = copts.encode('utf8')
     cdef const char *Src = <const char*>bSrc
@@ -741,6 +747,12 @@ def create_program_from_spirv (SyclQueue q, const unsigned char[:] IL):
             program (SyclProgram): A SyclProgram object wrapping the
                                    syc::program returned by the C API.
     '''
+    BE = q.get_sycl_backend()
+    if BE != backend_type.opencl:
+        raise ValueError(
+            "Cannot create program for a ", BE, "type backend. Currently only "
+            "OpenCL devices are supported for program creations."
+        )
 
     cdef DPPLSyclProgramRef Pref
     cdef const unsigned char *dIL = &IL[0]

dpctl/tests/test_sycl_program.py

@@ -81,6 +81,48 @@ def test_create_program_from_spirv(self):
                 self.assertEqual(addKernel.get_num_args(), 3)
                 self.assertEqual(axpyKernel.get_num_args(), 4)
 
+@unittest.skipUnless(
+    dpctl.has_gpu_queues(backend_ty=dpctl.backend_type.level_zero),
+    "No Level0 GPU queues available"
+)
+class TestProgramForLevel0GPU (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()
+            with dpctl.device_context("level0:gpu:0"):
+                q = dpctl.get_current_queue()
+                try:
+                    prog = dpctl.create_program_from_spirv(q,spirv)
+                    self.fail(
+                        "Tried to create program for an unsupported Level0 GPU."
+                    )
+                except ValueError:
+                    pass
+
+    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];                                  \
+        }"
+        with dpctl.device_context("level0:gpu:0"):
+            q = dpctl.get_current_queue()
+            try:
+                prog = dpctl.create_program_from_source(q, oclSrc)
+                self.fail(
+                    "Tried to create program for an unsupported Level0 GPU."
+                )
+            except ValueError:
+                pass
+
 
 if __name__ == '__main__':
     unittest.main()

dpctl/tests/test_sycl_queue.py

@@ -25,17 +25,25 @@
 import dpctl
 import unittest
 
-@unittest.skipUnless(dpctl.has_gpu_queues(), "No OpenCL GPU queues available")
-@unittest.skipUnless(dpctl.has_cpu_queues(), "No OpenCL GPU queues available")
 class TestSyclQueue (unittest.TestCase):
+    @unittest.skipUnless(
+        dpctl.has_gpu_queues(), "No OpenCL GPU queues available"
+    )
+    @unittest.skipUnless(
+        dpctl.has_cpu_queues(), "No OpenCL CPU queues available"
+    )
+    def test_queue_not_equals (self):
+        with dpctl.device_context("opencl:gpu") as gpuQ0:
+            with dpctl.device_context("opencl:cpu") as cpuQ:
+                self.assertFalse(cpuQ.equals(gpuQ0))
 
+    @unittest.skipUnless(
+        dpctl.has_gpu_queues(), "No OpenCL GPU queues available"
+    )
     def test_queue_equals (self):
         with dpctl.device_context("opencl:gpu") as gpuQ0:
             with dpctl.device_context("opencl:gpu") as gpuQ1:
                 self.assertTrue(gpuQ0.equals(gpuQ1))
-            with dpctl.device_context("opencl:cpu") as cpuQ:
-                self.assertFalse(cpuQ.equals(gpuQ0))
-
 
 if __name__ == '__main__':
     unittest.main()