Skip to content

Commit 784fa4b

Browse files
committed
tools(opset_coverage): Map default ops to unoverloaded ops
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent d630a1e commit 784fa4b

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

py/torch_tensorrt/dynamo/tools/opset_coverage.py

+11-3
Original file line numberDiff line numberDiff line change
@@ -122,14 +122,20 @@ def opset_coverage(
122122
converter_registry if converter_registry is not None else DYNAMO_CONVERTERS
123123
)
124124
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", "")
126128
for target in c_registry.keys()
127129
}
128130
supported_converted_targets = opset_targets.intersection(converter_registry_targets)
129131
support_count = 0
130132
legacy_count = 0
131133
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+
)
133139
if target_str in opset_targets:
134140
_, registry_data = c_registry.get_all_converters_with_target(
135141
target, return_registry_info=True
@@ -155,7 +161,9 @@ def opset_coverage(
155161
else get_decompositions()
156162
)
157163
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", "")
159167
for target in l_registry.keys()
160168
}
161169
supported_decomp_targets = opset_targets.intersection(decomp_registry_targets)

0 commit comments

Comments
 (0)