colossalai/kernel/triton - hpcaitech/ColossalAI - Morph

SIGN IN SIGN UP

hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

0 0 0 Python

	__init__.py	1.1 KB
	context_attn_unpad.py	26.1 KB
	flash_decoding.py	19.9 KB
	fused_rotary_embedding.py	4.7 KB
	kvcache_copy.py	10.5 KB
	llama_act_combine_kernel.py	6.7 KB
	no_pad_rotary_embedding.py	20.3 KB
	qkv_matmul_kernel.py	4.9 KB
	rms_layernorm.py	4.4 KB
	rotary_cache_copy.py	4.8 KB
	softmax.py	3.7 KB