github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-20 04:32:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove outdated triton test

Browse Source
This commit is contained in:
Yuanheng 2024-04-08 17:00:55 +08:00
parent d78817539e
commit 7ca1d1c545
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/kernel/triton/__init__.py
View File

@ -11,7 +11,6 @@ if HAS_TRITON:
from .context_attn_unpad import context_attention_unpadded from .context_attn_unpad import context_attention_unpadded
from .flash_decoding import flash_decoding_attention from .flash_decoding import flash_decoding_attention
from .fused_rotary_embedding import fused_rotary_embedding from .fused_rotary_embedding import fused_rotary_embedding
from .gptq_triton import gptq_fused_linear_triton
from .kvcache_copy import copy_kv_to_blocked_cache from .kvcache_copy import copy_kv_to_blocked_cache
from .no_pad_rotary_embedding import decoding_fused_rotary_embedding, rotary_embedding from .no_pad_rotary_embedding import decoding_fused_rotary_embedding, rotary_embedding
from .rms_layernorm import rms_layernorm from .rms_layernorm import rms_layernorm
@ -24,7 +23,6 @@ if HAS_TRITON:
"copy_kv_to_blocked_cache", "copy_kv_to_blocked_cache",
"softmax", "softmax",
"rms_layernorm", "rms_layernorm",
"gptq_fused_linear_triton",
"rotary_embedding", "rotary_embedding",
"fused_rotary_embedding", "fused_rotary_embedding",
"get_xine_cache", "get_xine_cache",