diff --git a/py/torch_tensorrt/dynamo/tools/opset_coverage.py b/py/torch_tensorrt/dynamo/tools/opset_coverage.py index 7e1560c855..9244f06f19 100644 --- a/py/torch_tensorrt/dynamo/tools/opset_coverage.py +++ b/py/torch_tensorrt/dynamo/tools/opset_coverage.py @@ -122,14 +122,20 @@ def opset_coverage( converter_registry if converter_registry is not None else DYNAMO_CONVERTERS ) converter_registry_targets = { - c_registry.qualified_name_or_str(target).removeprefix("torch.ops.") + c_registry.qualified_name_or_str(target) + .removeprefix("torch.ops.") + .replace(".default", "") for target in c_registry.keys() } supported_converted_targets = opset_targets.intersection(converter_registry_targets) support_count = 0 legacy_count = 0 for target in c_registry.keys(): - target_str = c_registry.qualified_name_or_str(target).removeprefix("torch.ops.") + target_str = ( + c_registry.qualified_name_or_str(target) + .removeprefix("torch.ops.") + .replace(".default", "") + ) if target_str in opset_targets: _, registry_data = c_registry.get_all_converters_with_target( target, return_registry_info=True @@ -155,7 +161,9 @@ def opset_coverage( else get_decompositions() ) decomp_registry_targets = { - c_registry.qualified_name_or_str(target).removeprefix("torch.ops.") + c_registry.qualified_name_or_str(target) + .removeprefix("torch.ops.") + .replace(".default", "") for target in l_registry.keys() } supported_decomp_targets = opset_targets.intersection(decomp_registry_targets)