use with_executor in remaining functions

NVIDIA · Dec 9, 2024 · 4da1966 · 4da1966
1 parent 0e9924b
commit 4da1966
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/python/test_transpose.py b/benchmarks/python/test_transpose.py
@@ -3,7 +3,7 @@
 # SPDX-License-Identifier: BSD-3-Clause
 import pytest
 from nvfuser import FusionDefinition, DataType
-from nvfuser.pytorch_utils import torch_dtype_to_nvfuser_dtype
+from nvfuser.pytorch_utils import torch_dtype_to_nvfuser_dtype, with_executor
 from .core import run_benchmark, clear_dynamo_cache
 import torch
 from .global_params import (