restore matmul tests

sony · Jan 8, 2025 · 3247949 · 3247949
1 parent c7af795
commit 3247949
Showing 1 changed file with 18 additions and 18 deletions.
diff --git a/tests/pytorch_tests/model_tests/test_feature_models_runner.py b/tests/pytorch_tests/model_tests/test_feature_models_runner.py
@@ -247,24 +247,24 @@ def test_linear_function(self):
         """
         LinearFNetTest(self).run_test()
 
-    # def test_matmul_function(self):
-    #     """
-    #     This test checks the MatMul substitution function
-    #     """
-    #     MatMulFNetTest(self, [3, 5, 10], [3, 10, 8]).run_test()
-    #     MatMulOpNetTest(self, [3, 5, 10], [3, 10, 8]).run_test()
-    #     MatMulFNetTest(self, [3, 2, 5, 10], [3, 2, 10, 20]).run_test()
-    #     MatMulOpNetTest(self, [3, 2, 5, 10], [3, 2, 10, 20]).run_test()
-    #     MatMulFNetTest(self, [50, 2, 400, 32], [50, 1, 32, 80]).run_test()
-    #     MatMulOpNetTest(self, [50, 2, 400, 32], [50, 1, 32, 80]).run_test()
-    #     MatMulFNetTest(self, [3, 1, 5, 10], [3, 8, 10, 3]).run_test()
-    #     MatMulOpNetTest(self, [3, 1, 5, 10], [3, 8, 10, 3]).run_test()
-    #     MatMulFNetTest(self, [3, 1, 4, 5, 10], [3, 8, 1, 10, 10]).run_test()
-    #     MatMulOpNetTest(self, [3, 1, 4, 5, 10], [3, 8, 1, 10, 10]).run_test()
-    #     MatMulFNetTest(self, [3, 10, 6, 5, 50, 100], [3, 10, 1, 1, 100, 80]).run_test()
-    #     MatMulOpNetTest(self, [3, 10, 6, 5, 50, 100], [3, 10, 1, 1, 100, 80]).run_test()
-    #     MatMulFNetTest(self, [3, 1, 7, 1, 50, 100], [3, 10, 7, 5, 100, 80]).run_test()
-    #     MatMulOpNetTest(self, [3, 1, 7, 1, 50, 100], [3, 10, 7, 5, 100, 80]).run_test()
+    def test_matmul_function(self):
+        """
+        This test checks the MatMul substitution function
+        """
+        MatMulFNetTest(self, [3, 5, 10], [3, 10, 8]).run_test()
+        MatMulOpNetTest(self, [3, 5, 10], [3, 10, 8]).run_test()
+        MatMulFNetTest(self, [3, 2, 5, 10], [3, 2, 10, 20]).run_test()
+        MatMulOpNetTest(self, [3, 2, 5, 10], [3, 2, 10, 20]).run_test()
+        MatMulFNetTest(self, [50, 2, 400, 32], [50, 1, 32, 80]).run_test()
+        MatMulOpNetTest(self, [50, 2, 400, 32], [50, 1, 32, 80]).run_test()
+        MatMulFNetTest(self, [3, 1, 5, 10], [3, 8, 10, 3]).run_test()
+        MatMulOpNetTest(self, [3, 1, 5, 10], [3, 8, 10, 3]).run_test()
+        MatMulFNetTest(self, [3, 1, 4, 5, 10], [3, 8, 1, 10, 10]).run_test()
+        MatMulOpNetTest(self, [3, 1, 4, 5, 10], [3, 8, 1, 10, 10]).run_test()
+        MatMulFNetTest(self, [3, 10, 6, 5, 50, 100], [3, 10, 1, 1, 100, 80]).run_test()
+        MatMulOpNetTest(self, [3, 10, 6, 5, 50, 100], [3, 10, 1, 1, 100, 80]).run_test()
+        MatMulFNetTest(self, [3, 1, 7, 1, 50, 100], [3, 10, 7, 5, 100, 80]).run_test()
+        MatMulOpNetTest(self, [3, 1, 7, 1, 50, 100], [3, 10, 7, 5, 100, 80]).run_test()
 
     def test_broken_net(self):
         """