Fix hanging in tril

IntelPython · Jun 9, 2023 · a8acc96 · a8acc96
1 parent c286b52
commit a8acc96
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/dpnp/backend/extensions/vm/div.cpp b/dpnp/backend/extensions/vm/div.cpp
@@ -219,6 +219,11 @@ bool can_call_div(sycl::queue exec_q,
     {
         return false;
     }
+    else if (dst_nd == 0)
+    {
+        // don't call OneMKL for 0d arrays
+        return false;
+    }
 
     // shapes must be the same
     const py::ssize_t* src1_shape = src1.get_shape_raw();