diff --git a/tests/pytorch_tests/model_tests/feature_models/matmul_test.py b/tests/pytorch_tests/model_tests/feature_models/matmul_test.py index 935570956..c457a9319 100644 --- a/tests/pytorch_tests/model_tests/feature_models/matmul_test.py +++ b/tests/pytorch_tests/model_tests/feature_models/matmul_test.py @@ -51,6 +51,7 @@ def __init__(self, unit_test, input_shape, other_shape): super().__init__(unit_test) self.input_shape = input_shape self.other_shape = other_shape + self.use_is_close_validation = True # There is a small difference between float operations def create_inputs_shape(self): return [self.input_shape, self.other_shape]