[NFC] Change xetla_kernel import style (#2199)

CI status: https://github.com/intel/intel-xpu-backend-for-triton/actions/runs/10809894211 Signed-off-by: Anatoly Myachev <[email protected]>
intel · Sep 11, 2024 · 3102cf5 · 3102cf5
1 parent 270c13a
commit 3102cf5
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/benchmarks/triton_kernels_benchmark/flash_attention_fwd_benchmark.py b/benchmarks/triton_kernels_benchmark/flash_attention_fwd_benchmark.py
@@ -5,7 +5,7 @@
 import triton.language as tl
 
 import triton_kernels_benchmark
-from triton_kernels_benchmark import xetla_kernel  # pylint: disable=no-name-in-module
+import xetla_kernel
 
 benchmark_suit = triton_kernels_benchmark  # triton.testing
 

diff --git a/benchmarks/triton_kernels_benchmark/fused_softmax.py b/benchmarks/triton_kernels_benchmark/fused_softmax.py
@@ -15,7 +15,7 @@
 from triton.runtime import driver
 
 import triton_kernels_benchmark
-from triton_kernels_benchmark import xetla_kernel  # pylint: disable=no-name-in-module
+import xetla_kernel
 
 benchmark_suit = triton_kernels_benchmark  # triton.testing
 

diff --git a/benchmarks/triton_kernels_benchmark/gemm_benchmark.py b/benchmarks/triton_kernels_benchmark/gemm_benchmark.py
@@ -14,7 +14,7 @@
 import triton.language as tl
 
 import triton_kernels_benchmark as benchmark_suit
-from triton_kernels_benchmark import xetla_kernel  # pylint: disable=no-name-in-module
+import xetla_kernel
 
 
 @triton.autotune(