Change org

pytorch-labs · Nov 26, 2024 · 6e6df10 · 6e6df10
1 parent 3bbc161
commit 6e6df10
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tritonbench/operators/op.py b/tritonbench/operators/op.py
@@ -67,7 +67,7 @@ def load_opbench_by_name(op_name: str):
         if not _is_internal_operator(op_name)
         else f"{INTERNAL_OPBENCH_DIR}.{op_name}"
     )
-    module = importlib.import_module(f".{op_pkg}", package=__name__)
+    module = importlib.import_module(f"..{op_pkg}", package=__name__)
 
     Operator = getattr(module, "Operator", None)
     if Operator is None:

diff --git a/tritonbench/operators_collection/op_collection.py b/tritonbench/operators_collection/op_collection.py
@@ -49,7 +49,7 @@ def list_operators_by_collection(op_collection: str = "default") -> List[str]:
 
     def _list_all_operators(collection_name: str):
         try:
-            module_name = f".{collection_name}"
+            module_name = f"..{collection_name}"
             module = importlib.import_module(module_name, package=__name__)
             if hasattr(module, "get_operators"):
                 return module.get_operators()