Load operators

pytorch-labs · Nov 26, 2024 · 164849d · 164849d
1 parent 6e6df10
commit 164849d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tritonbench/__init__.py b/tritonbench/__init__.py
@@ -1,2 +1,2 @@
-from . import operators
-from . import operators_collection
+from .operators import list_operators, load_opbench_by_name
+from .operators_collection import list_operator_collections, list_operators_by_collection