tools: Opset coverage notebook

Signed-off-by: Naren Dasan <naren@narendasan.com> Signed-off-by: Naren Dasan <narens@nvidia.com>
pytorch · May 13, 2024 · a811c4a · a811c4a
1 parent 4dc9acf
commit a811c4a
Show file tree

Hide file tree

Showing 3 changed files with 616 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -69,3 +69,8 @@ bazel-tensorrt
 bazel-project
 build/
 wheelhouse/
+*_status.json
+tests/py/dynamo/models/*.ts
+tests/py/dynamo/models/*.ep
+*.deb
+*.tar.xz
diff --git a/py/torch_tensorrt/dynamo/tools/opset_coverage.py b/py/torch_tensorrt/dynamo/tools/opset_coverage.py
@@ -206,6 +206,15 @@ def opset_coverage(
     )
 
 
+def get_coverage_status(opset: List[Tuple[str, str]], name: str) -> OpsetCoverage:
+    coverage = opset_coverage(opset)
+    return coverage
+
+
+ATEN_COVERAGE = get_coverage_status(ATEN_OPS, "ATen")
+PRIMS_COVERAGE = get_coverage_status(PRIM_OPS, "prim")
+PY_OVERLOAD_COVERAGE = get_coverage_status(OVERLOADED_PY_OPS, "py_overload")
+
 if __name__ == "__main__":
 
     def find_coverage_status(opset: List[Tuple[str, str]], name: str) -> None: