Fix hanging in tril

antonwolfy · antonwolfy · commit a8acc96be49f · 2023-06-09T12:02:08.000-05:00
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();

Original file line number	Diff line number	Diff line change
`@@ -219,6 +219,11 @@ bool can_call_div(sycl::queue exec_q,`
`219`	`219`	`{`
`220`	`220`	`return false;`
`221`	`221`	`}`
	`222`	`+ else if (dst_nd == 0)`
	`223`	`+ {`
	`224`	`+ // don't call OneMKL for 0d arrays`
	`225`	`+ return false;`
	`226`	`+ }`
`222`	`227`
`223`	`228`	`// shapes must be the same`
`224`	`229`	`const py::ssize_t* src1_shape = src1.get_shape_raw();`