File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed
py/torch_tensorrt/dynamo/tools Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change @@ -129,7 +129,11 @@ def opset_coverage(
129
129
support_count = 0
130
130
legacy_count = 0
131
131
for target in c_registry .keys ():
132
- target_str = c_registry .qualified_name_or_str (target ).removeprefix ("torch.ops." )
132
+ target_str = (
133
+ c_registry .qualified_name_or_str (target )
134
+ .removeprefix ("torch.ops." )
135
+ .replace (".default" , "" )
136
+ )
133
137
if target_str in opset_targets :
134
138
_ , registry_data = c_registry .get_all_converters_with_target (
135
139
target , return_registry_info = True
@@ -155,7 +159,9 @@ def opset_coverage(
155
159
else get_decompositions ()
156
160
)
157
161
decomp_registry_targets = {
158
- c_registry .qualified_name_or_str (target ).removeprefix ("torch.ops." )
162
+ c_registry .qualified_name_or_str (target )
163
+ .removeprefix ("torch.ops." )
164
+ .replace (".default" , "" )
159
165
for target in l_registry .keys ()
160
166
}
161
167
supported_decomp_targets = opset_targets .intersection (decomp_registry_targets )
You can’t perform that action at this time.
0 commit comments