diff --git a/tests/pytorch_tests/model_tests/test_feature_models_runner.py b/tests/pytorch_tests/model_tests/test_feature_models_runner.py index 9e9d334ae..090bbe713 100644 --- 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): """