Skip to content

Commit d56025d

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 d56025d

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

py/torch_tensorrt/dynamo/tools/opset_coverage.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -129,7 +129,11 @@ def opset_coverage(
129129
support_count = 0
130130
legacy_count = 0
131131
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+
)
133137
if target_str in opset_targets:
134138
_, registry_data = c_registry.get_all_converters_with_target(
135139
target, return_registry_info=True
@@ -155,7 +159,9 @@ def opset_coverage(
155159
else get_decompositions()
156160
)
157161
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", "")
159165
for target in l_registry.keys()
160166
}
161167
supported_decomp_targets = opset_targets.intersection(decomp_registry_targets)

0 commit comments

Comments
 (0)