Adjusted tests to work on GPUs w/o double precision support

Author: oleksandr-pavlyk

examples/pybind11/onemkl_gemv/sycl_gemm/_onemkl.cpp

@@ -62,13 +62,11 @@ py_gemv(sycl::queue q,
         throw std::runtime_error("Inconsistent shapes.");
     }
 
-    auto q_ctx = q.get_context();
-    if (q_ctx != matrix.get_queue().get_context() ||
-        q_ctx != vector.get_queue().get_context() ||
-        q_ctx != result.get_queue().get_context())
+    if (!dpctl::utils::queues_are_compatible(
+            q, {matrix.get_queue(), vector.get_queue(), result.get_queue()}))
     {
         throw std::runtime_error(
-            "USM allocation is not bound to the context in execution queue.");
+            "USM allocations are not compatible with the execution queue.");
     }
 
     auto &api = dpctl::detail::dpctl_capi::get();

examples/pybind11/onemkl_gemv/tests/test_gemm.py

@@ -18,9 +18,9 @@ def test_gemv():
     except dpctl.SyclQueueCreationError:
         pytest.skip("Queue could not be created")
     Mnp, vnp = np.random.randn(5, 3), np.random.randn(3)
-    r = dpt.empty((5,), dtype="d", sycl_queue=q)
     M = dpt.asarray(Mnp, sycl_queue=q)
     v = dpt.asarray(vnp, sycl_queue=q)
+    r = dpt.empty((5,), dtype=v.dtype, sycl_queue=q)
     hev, ev = gemv(q, M, v, r, [])
     hev.wait()
     rnp = dpt.asnumpy(r)
@@ -33,9 +33,9 @@ def test_sub():
     except dpctl.SyclQueueCreationError:
         pytest.skip("Queue could not be created")
     anp, bnp = np.random.randn(5), np.random.randn(5)
-    r = dpt.empty((5,), dtype="d", sycl_queue=q)
     a = dpt.asarray(anp, sycl_queue=q)
     b = dpt.asarray(bnp, sycl_queue=q)
+    r = dpt.empty((5,), dtype=b.dtype, sycl_queue=q)
     hev, ev = sub(q, a, b, r, [])
     hev.wait()
     rnp = dpt.asnumpy(r)