From 164849db07d894a4d457c916abac93fbe86f1bf1 Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Tue, 26 Nov 2024 13:49:58 -0500 Subject: [PATCH] Load operators --- tritonbench/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tritonbench/__init__.py b/tritonbench/__init__.py index 3186c98e..617daa4e 100644 --- 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