@@ -122,14 +122,20 @@ def opset_coverage(
122
122
converter_registry if converter_registry is not None else DYNAMO_CONVERTERS
123
123
)
124
124
converter_registry_targets = {
125
- c_registry .qualified_name_or_str (target ).removeprefix ("torch.ops." )
125
+ c_registry .qualified_name_or_str (target )
126
+ .removeprefix ("torch.ops." )
127
+ .replace (".default" , "" )
126
128
for target in c_registry .keys ()
127
129
}
128
130
supported_converted_targets = opset_targets .intersection (converter_registry_targets )
129
131
support_count = 0
130
132
legacy_count = 0
131
133
for target in c_registry .keys ():
132
- target_str = c_registry .qualified_name_or_str (target ).removeprefix ("torch.ops." )
134
+ target_str = (
135
+ c_registry .qualified_name_or_str (target )
136
+ .removeprefix ("torch.ops." )
137
+ .replace (".default" , "" )
138
+ )
133
139
if target_str in opset_targets :
134
140
_ , registry_data = c_registry .get_all_converters_with_target (
135
141
target , return_registry_info = True
@@ -155,7 +161,9 @@ def opset_coverage(
155
161
else get_decompositions ()
156
162
)
157
163
decomp_registry_targets = {
158
- c_registry .qualified_name_or_str (target ).removeprefix ("torch.ops." )
164
+ c_registry .qualified_name_or_str (target )
165
+ .removeprefix ("torch.ops." )
166
+ .replace (".default" , "" )
159
167
for target in l_registry .keys ()
160
168
}
161
169
supported_decomp_targets = opset_targets .intersection (decomp_registry_targets )
0 commit comments