diff --git a/benchmarks/triton_kernels_benchmark/flash_attention_fwd_benchmark.py b/benchmarks/triton_kernels_benchmark/flash_attention_fwd_benchmark.py
index 8863483fda..f1a0d1b6ab 100644
--- 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
index 1586033235..0fd95f6f2f 100644
--- 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
index 2cc4071491..f8f390dcf3 100644
--- 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(