From 324794940de65c4a9426c9e56eac1e860c42ee05 Mon Sep 17 00:00:00 2001 From: irenab Date: Wed, 8 Jan 2025 12:39:55 +0200 Subject: [PATCH] restore matmul tests --- .../model_tests/test_feature_models_runner.py | 36 +++++++++---------- 1 file changed, 18 insertions(+), 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 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): """