tools: Opset coverage notebook

narendasan · narendasan · commit a811c4a3869a · 2024-05-13T16:52:33.000-07:00
Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
Signed-off-by: Naren Dasan &lt;narens@nvidia.com&gt;
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:
diff --git a/tools/opset_coverage.ipynb b/tools/opset_coverage.ipynb
