File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed
tests/py/dynamo/conversion Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -267,7 +267,7 @@ def run_test(
267267 check_dtype ,
268268 )
269269
270- def run_test_comparator (
270+ def run_test_compare_tensor_attributes_only (
271271 self ,
272272 mod ,
273273 inputs ,
Original file line number Diff line number Diff line change @@ -86,7 +86,7 @@ def forward(self, x):
8686 x .dtype == y .dtype if check_dtype else True
8787 )
8888 expected_ops = []
89- self .run_test_comparator (
89+ self .run_test_compare_tensor_attributes_only (
9090 rand_model ,
9191 inputs ,
9292 expected_ops ,
@@ -122,7 +122,7 @@ def forward(self, x):
122122 )
123123 expected_ops = []
124124 # TRT-TRT returns int32 while torch returns int64
125- self .run_test_comparator (
125+ self .run_test_compare_tensor_attributes_only (
126126 rand_model ,
127127 inputs ,
128128 expected_ops ,
You can’t perform that action at this time.
0 commit comments