diff --git a/tests/test_infer_ops/triton/utils.py b/tests/test_infer_ops/triton/kernel_utils.py
similarity index 100%
rename from tests/test_infer_ops/triton/utils.py
rename to tests/test_infer_ops/triton/kernel_utils.py
diff --git a/tests/test_infer_ops/triton/test_bloom_context_attention.py b/tests/test_infer_ops/triton/test_bloom_context_attention.py
index 7447c85c5887..344ad078e2e2 100644
--- a/tests/test_infer_ops/triton/test_bloom_context_attention.py
+++ b/tests/test_infer_ops/triton/test_bloom_context_attention.py
@@ -11,7 +11,7 @@
     import triton.language as tl
 
     from colossalai.kernel.triton import bloom_context_attn_fwd
-    from tests.test_infer_ops.triton.utils import torch_context_attention
+    from tests.test_infer_ops.triton.kernel_utils import torch_context_attention
     HAS_TRITON = True
 except ImportError:
     HAS_TRITON = False
diff --git a/tests/test_infer_ops/triton/test_llama_context_attention.py b/tests/test_infer_ops/triton/test_llama_context_attention.py
index 1659fdde8f7f..4ea6095d4109 100644
--- a/tests/test_infer_ops/triton/test_llama_context_attention.py
+++ b/tests/test_infer_ops/triton/test_llama_context_attention.py
@@ -11,7 +11,7 @@
     import triton.language as tl
 
     from colossalai.kernel.triton import llama_context_attn_fwd
-    from tests.test_infer_ops.triton.utils import torch_context_attention
+    from tests.test_infer_ops.triton.kernel_utils import torch_context_attention
     HAS_TRITON = True
 except ImportError:
     HAS_TRITON = False