Change name of run_test_comparator function in harness.py

pytorch · Apr 16, 2024 · c3c768c · c3c768c
1 parent c76c2c3
commit c3c768c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/py/dynamo/conversion/harness.py b/tests/py/dynamo/conversion/harness.py
@@ -251,7 +251,7 @@ def run_test(
             check_dtype,
         )
 
-    def run_test_comparator(
+    def run_test_compare_tensor_attributes_only(
         self,
         mod,
         inputs,

diff --git a/tests/py/dynamo/conversion/test_rand_aten.py b/tests/py/dynamo/conversion/test_rand_aten.py
@@ -86,7 +86,7 @@ def forward(self, x):
             x.dtype == y.dtype if check_dtype else True
         )
         expected_ops = []
-        self.run_test_comparator(
+        self.run_test_compare_tensor_attributes_only(
             rand_model,
             inputs,
             expected_ops,
@@ -122,7 +122,7 @@ def forward(self, x):
         )
         expected_ops = []
         # TRT-TRT returns int32  while torch returns int64
-        self.run_test_comparator(
+        self.run_test_compare_tensor_attributes_only(
             rand_model,
             inputs,
             expected_ops,
-Original file line number
+Diff line change
@@ Expand Up / @@ -251,7 +251,7 @@ def run_test( @@
                 check_dtype,
             )
-        def run_test_comparator(
+        def run_test_compare_tensor_attributes_only(
             self,
             mod,
             inputs,
@@ Expand Down @@