From 7ecd74e79ef668842a65984e209fece0fb65ebbe Mon Sep 17 00:00:00 2001 From: Anatoly Myachev Date: Wed, 11 Sep 2024 10:31:19 +0000 Subject: [PATCH] [NFC] Change 'xetla_kernel' import style Signed-off-by: Anatoly Myachev --- .../triton_kernels_benchmark/flash_attention_fwd_benchmark.py | 2 +- benchmarks/triton_kernels_benchmark/fused_softmax.py | 2 +- benchmarks/triton_kernels_benchmark/gemm_benchmark.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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(