Zhenghai Zhang c46d5e48f8 【Hackathon 9th No.86】autogen MultiQueryAppendC8Attention template_instantiation -part (#4330) ...

* split MultiQueryAppendC8Attention template_instantiation

* update setup_ops.py

* fix ci

* fix bug

2025-10-10 15:07:48 +08:00

..

template_instantiation

【Hackathon 9th No.86】autogen MultiQueryAppendC8Attention template_instantiation -part (#4330)

2025-10-10 15:07:48 +08:00

append_attention_c4_impl.cuh

[Excutor] Experiment Feature-Support Prefill in cudagraph (#3459)

2025-09-08 13:12:24 +08:00

append_attention_c8_impl.cuh

【Hackathon 9th No.86】autogen MultiQueryAppendC8Attention template_instantiation -part (#4330)

2025-10-10 15:07:48 +08:00

append_attention_c16_impl.cuh

[Excutor] Experiment Feature-Support Prefill in cudagraph (#3459)

2025-09-08 13:12:24 +08:00

append_attention_func.cuh

[Executor]CUDAGraph support Speculate Decode (#3769)

2025-10-09 21:18:29 +08:00

append_attention_kernel.h

【Hackathon 9th No.86】autogen MultiQueryAppendC8Attention template_instantiation -part (#4330)

2025-10-10 15:07:48 +08:00

autogen_template_instantiation.py

【Hackathon 9th No.86】autogen MultiQueryAppendC8Attention template_instantiation -part (#4330)

2025-10-10 15:07:48 +08:00

decode_attention_func.cuh

[Sync] Update to latest code (#2679)

2025-07-03 15:43:53 +08:00

decode_attention_kernel.cu

[Inference, rename] remove padding_offsets from atten use batch_id_per_token (#2880)

2025-07-17 18:41:31 +08:00

decoder_write_cache_with_rope_impl.cuh

fix dynamic Cfp8 computing error (#4119)

2025-09-16 20:21:49 +08:00

decoder_write_cache_with_rope_kernel.cu

[Feature] Support zai-org/GLM-4.5-Air BF16 model (#3928)

2025-09-10 19:36:10 +08:00

decoder_write_cache_with_rope_kernel.h

clean code in sttantion (#3917)

2025-09-05 20:49:01 +08:00

encoder_write_cache_with_rope_impl.cuh

FIX] Fix CUDA error(700): 'cudaErrorIllegalAddress' in CascadeAppendWriteCacheKVQKV cache_kernel(). Continue when batch_id_per_token[token_idx] is default value -1. (#4218)

2025-09-24 14:08:49 +08:00

encoder_write_cache_with_rope_kernel.h

[Feature] Support zai-org/GLM-4.5-Air BF16 model (#3928)

2025-09-10 19:36:10 +08:00

get_block_shape_and_split_kv_block.cu

【Inference Optimize】MLA Tensor-Core is enabled by default (#4335)

2025-10-10 10:54:56 +08:00

gqa_rope_write_cache.cu

fix typos (#3951)

2025-09-08 15:22:41 +08:00

mem_util.cuh

[LLM] First commit the llm deployment code

2025-06-09 19:20:15 +08:00

mla_cache_kernel.cu

[Feature] DeepseekV3 use pd_build_static_op (#2948)

2025-07-22 15:03:41 +08:00

mla_cache_kernel.cuh

[Inference, rename] remove padding_offsets from atten use batch_id_per_token (#2880)

2025-07-17 18:41:31 +08:00

mma_tensor_op.cuh

[LLM] First commit the llm deployment code

2025-06-09 19:20:15 +08:00

multi_head_latent_attention_kernel.h

[Inference, rename] remove padding_offsets from atten use batch_id_per_token (#2880)

2025-07-17 18:41:31 +08:00

multiquery_attention_c8_impl.cuh

【Hackathon 9th No.86】autogen MultiQueryAppendC8Attention template_instantiation -part (#4330)

2025-10-10 15:07:48 +08:00

multiquery_attention_c8_kernel.h

【Hackathon 9th No.86】autogen MultiQueryAppendC8Attention template_instantiation -part (#4330)

2025-10-10 15:07:48 +08:00

pre_cache_len_concat.cu

[LLM] First commit the llm deployment code

2025-06-09 19:20:15 +08:00

speculate_write_cache_with_rope_impl.cuh

[Executor]CUDAGraph support Speculate Decode (#3769)

2025-10-09 21:18:29 +08:00

speculate_write_cache_with_rope_kernel.cu

support rope_3d in spec mode (#4034)

2025-09-10 03:15:05 -07:00

speculate_write_cache_with_rope_kernel.h

support mtp rope_3d (#3791)

2025-09-04 17:18:05 +08:00

utils.cuh

supports dynamic Cfp8 (#3767)

2025-09-07 20:41:29 -07:00