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(
251
251
check_dtype ,
252
252
)
253
253
254
- def run_test_comparator (
254
+ def run_test_compare_tensor_attributes_only (
255
255
self ,
256
256
mod ,
257
257
inputs ,
Original file line number Diff line number Diff line change @@ -86,7 +86,7 @@ def forward(self, x):
86
86
x .dtype == y .dtype if check_dtype else True
87
87
)
88
88
expected_ops = []
89
- self .run_test_comparator (
89
+ self .run_test_compare_tensor_attributes_only (
90
90
rand_model ,
91
91
inputs ,
92
92
expected_ops ,
@@ -122,7 +122,7 @@ def forward(self, x):
122
122
)
123
123
expected_ops = []
124
124
# TRT-TRT returns int32 while torch returns int64
125
- self .run_test_comparator (
125
+ self .run_test_compare_tensor_attributes_only (
126
126
rand_model ,
127
127
inputs ,
128
128
expected_ops ,
You can’t perform that action at this time.
0 commit comments