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 @@ -251,7 +251,7 @@ def run_test(
251251 check_dtype ,
252252 )
253253
254- def run_test_comparator (
254+ def run_test_compare_tensor_attributes_only (
255255 self ,
256256 mod ,
257257 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