Skip to content

Commit be9aa67

Browse files
committed
Support FX Legacy registry name; use constants for registry names
1 parent 22cf701 commit be9aa67

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

py/torch_tensorrt/dynamo/tools/opset_coverage.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,10 @@ class OpsetCoverage:
4848
Path(os.path.dirname(torchgen.__file__)) / "packaged/ATen/native/tags.yaml"
4949
)
5050

51+
DYNAMO_REGISTRY_NAME = "Dynamo ATen Converters Registry"
52+
FX_REGISTRY_NAME = "FX ATen Converters Registry"
53+
FX_LEGACY_REGISTRY_NAME = "FX Legacy ATen Converters Registry"
54+
5155

5256
def get_aten_ops() -> List[Tuple[str, str]]:
5357
parsed_yaml = parse_native_yaml(NATIVE_FUNCTION_YAML_PATH, TAGS_YAML_PATH)
@@ -140,11 +144,12 @@ def opset_coverage(
140144
_, registry_data = c_registry.get_all_converters_with_target(
141145
target, return_registry_info=True
142146
)
147+
143148
if registry_data is not None:
144-
if registry_data["Dynamo ATen Converters Registry"] >= 1:
149+
if DYNAMO_REGISTRY_NAME in registry_data and registry_data[DYNAMO_REGISTRY_NAME] >= 1:
145150
status = SupportStatus.CONVERTED
146151
support_count += 1
147-
elif registry_data["FX ATen Converters Registry"] >= 1:
152+
elif (FX_REGISTRY_NAME in registry_data and registry_data[FX_REGISTRY_NAME] >= 1) or (FX_LEGACY_REGISTRY_NAME in registry_data and registry_data[FX_LEGACY_REGISTRY_NAME] >= 1):
148153
status = SupportStatus.LEGACY_CONVERTED
149154
legacy_count += 1
150155

0 commit comments

Comments
 (0)