Enable tests for matmul

Author: antonwolfy

dpnp/tests/test_usm_type.py

@@ -458,45 +458,45 @@ def test_coerced_usm_types_bitwise_op(op, usm_type_x, usm_type_y):
     assert z.usm_type == du.get_coerced_usm_type([usm_type_x, usm_type_y])
 
 
-# @pytest.mark.parametrize("usm_type_x", list_of_usm_types, ids=list_of_usm_types)
-# @pytest.mark.parametrize("usm_type_y", list_of_usm_types, ids=list_of_usm_types)
-# @pytest.mark.parametrize(
-#     "shape_pair",
-#     [
-#         ((2, 4), (4,)),
-#         ((4,), (4, 3)),
-#         ((2, 4), (4, 3)),
-#         ((2, 0), (0, 3)),
-#         ((2, 4), (4, 0)),
-#         ((4, 2, 3), (4, 3, 5)),
-#         ((4, 2, 3), (4, 3, 1)),
-#         ((4, 1, 3), (4, 3, 5)),
-#         ((6, 7, 4, 3), (6, 7, 3, 5)),
-#     ],
-#     ids=[
-#         "((2, 4), (4,))",
-#         "((4,), (4, 3))",
-#         "((2, 4), (4, 3))",
-#         "((2, 0), (0, 3))",
-#         "((2, 4), (4, 0))",
-#         "((4, 2, 3), (4, 3, 5))",
-#         "((4, 2, 3), (4, 3, 1))",
-#         "((4, 1, 3), (4, 3, 5))",
-#         "((6, 7, 4, 3), (6, 7, 3, 5))",
-#     ],
-# )
-# def test_matmul(usm_type_x, usm_type_y, shape_pair):
-#     shape1, shape2 = shape_pair
-#     x = numpy.arange(numpy.prod(shape1)).reshape(shape1)
-#     y = numpy.arange(numpy.prod(shape2)).reshape(shape2)
+@pytest.mark.parametrize("usm_type_x", list_of_usm_types, ids=list_of_usm_types)
+@pytest.mark.parametrize("usm_type_y", list_of_usm_types, ids=list_of_usm_types)
+@pytest.mark.parametrize(
+    "shape_pair",
+    [
+        ((2, 4), (4,)),
+        ((4,), (4, 3)),
+        ((2, 4), (4, 3)),
+        ((2, 0), (0, 3)),
+        ((2, 4), (4, 0)),
+        ((4, 2, 3), (4, 3, 5)),
+        ((4, 2, 3), (4, 3, 1)),
+        ((4, 1, 3), (4, 3, 5)),
+        ((6, 7, 4, 3), (6, 7, 3, 5)),
+    ],
+    ids=[
+        "((2, 4), (4,))",
+        "((4,), (4, 3))",
+        "((2, 4), (4, 3))",
+        "((2, 0), (0, 3))",
+        "((2, 4), (4, 0))",
+        "((4, 2, 3), (4, 3, 5))",
+        "((4, 2, 3), (4, 3, 1))",
+        "((4, 1, 3), (4, 3, 5))",
+        "((6, 7, 4, 3), (6, 7, 3, 5))",
+    ],
+)
+def test_matmul(usm_type_x, usm_type_y, shape_pair):
+    shape1, shape2 = shape_pair
+    x = numpy.arange(numpy.prod(shape1)).reshape(shape1)
+    y = numpy.arange(numpy.prod(shape2)).reshape(shape2)
 
-#     x = dp.array(x, usm_type=usm_type_x)
-#     y = dp.array(y, usm_type=usm_type_y)
-#     z = dp.matmul(x, y)
+    x = dp.array(x, usm_type=usm_type_x)
+    y = dp.array(y, usm_type=usm_type_y)
+    z = dp.matmul(x, y)
 
-#     assert x.usm_type == usm_type_x
-#     assert y.usm_type == usm_type_y
-#     assert z.usm_type == du.get_coerced_usm_type([usm_type_x, usm_type_y])
+    assert x.usm_type == usm_type_x
+    assert y.usm_type == usm_type_y
+    assert z.usm_type == du.get_coerced_usm_type([usm_type_x, usm_type_y])
 
 
 # @pytest.mark.parametrize("usm_type_x", list_of_usm_types, ids=list_of_usm_types)