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(
267
267
check_dtype ,
268
268
)
269
269
270
- def run_test_comparator (
270
+ def run_test_compare_tensor_attributes_only (
271
271
self ,
272
272
mod ,
273
273
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