Fix hanging in tril

antonwolfy · antonwolfy · commit ee09d7b8740c · 2023-06-09T12:04:48.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();
diff --git a/tests/test_mathematical.py b/tests/test_mathematical.py
@@ -284,7 +284,7 @@ def test_divide_scalar(shape, dtype):
 
     result = 0.5 / dpnp_a / 1.7
     expected = 0.5 / np_a / 1.7
-    assert_allclose(result, expected)
+    assert_allclose(result, expected, rtol=1e-6)
 
 
 @pytest.mark.parametrize("shape",

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();`