fix import error

NVIDIA · Dec 10, 2024 · 3fa27fb · 3fa27fb
1 parent 810fcb3
commit 3fa27fb
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/benchmarks/python/test_transpose.py b/benchmarks/python/test_transpose.py
@@ -3,15 +3,10 @@
 # SPDX-License-Identifier: BSD-3-Clause
 import pytest
 from nvfuser import FusionDefinition, DataType
-from nvfuser.pytorch_utils import torch_dtype_to_nvfuser_dtype, with_executor
-from .core import run_benchmark, clear_dynamo_cache
+from nvfuser.pytorch_utils import torch_dtype_to_nvfuser_dtype
+from .core import run_benchmark, clear_dynamo_cache, with_executor, DEFAULT_EXECUTORS
 import torch
-from .global_params import (
-    generate_input_sizes,
-    FLOAT_DTYPES,
-    PROMOTE_DTYPES,
-    with_executor,
-)
+from .global_params import generate_input_sizes, FLOAT_DTYPES, PROMOTE_DTYPES
 
 
 def transpose_fusion(