From 6654ade07885fec56cb2e42a3694829ba9f5f2a3 Mon Sep 17 00:00:00 2001 From: yuanheng-zhao Date: Thu, 7 Sep 2023 18:01:19 +0800 Subject: [PATCH] modify utils filename for infer ops test --- tests/test_infer_ops/triton/{utils.py => kernel_utils.py} | 0 tests/test_infer_ops/triton/test_bloom_context_attention.py | 2 +- tests/test_infer_ops/triton/test_llama_context_attention.py | 2 +- 3 files changed, 2 insertions(+), 2 deletions(-) rename tests/test_infer_ops/triton/{utils.py => kernel_utils.py} (100%) 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