FastDeploy

append_attn

[Speculative Decoding][MTP] Support static CacheKV C8 quantization and optimize memory usage (#5155 )

2025-11-21 15:10:13 +08:00

common

polish code with new pre-commit rule (#2923 )

2025-07-19 23:19:27 +08:00

custom_all_reduce

[NewFeature]custom_allreduce support cudagraph recapture (#4305 )

2025-09-29 15:56:54 +08:00

cutlass_extensions

fix typos (#3684 )

2025-09-01 17:50:17 +08:00

cutlass_kernels

fix typos (#3684 )

2025-09-01 17:50:17 +08:00

flash_mask_attn

format flash_mask_attn

2025-11-18 17:18:12 +08:00

fp8_gemm_with_cutlass

polish code with new pre-commit rule (#2923 )

2025-07-19 23:19:27 +08:00

glog

[LLM] First commit the llm deployment code

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

int8_gemm_with_cutlass

fix typos (#3684 )

2025-09-01 17:50:17 +08:00

machete

WINT4/WINT8 dense gemm default use Machete (#4451 )

2025-10-23 17:57:59 +08:00

mla_attn

【Inference Optimize】DeepSeek-V3-model MLA Optimize (#3886 )

2025-09-11 10:46:09 +08:00

moba_attn

enhance set_stop_value_multi_ends and standardize the registration of some operators (#4525 )

2025-10-21 22:06:06 +08:00

moe

[BugFix] Fix zero workspace returned by CUB size query under CUDA Graph in MoE dispatch (#5087 )

2025-11-20 20:00:29 +08:00

quantization

[OPs] MoE support wfp8afp8(channelwise) and improve per_token_quant_fp8 (#4238 )

2025-09-24 16:39:51 +08:00

sample_kernels

[OPs] Universal optimization and Fix early_stop cuda 700 (#3375 )

2025-08-14 22:40:44 +08:00

speculate_decoding

[BugFix] fix mtp logprob bugs in chunk prefill (#5244 )

2025-11-27 11:31:29 +08:00

w4afp8_gemm

【New Feature】W4afp8 supports per group quantization (#4987 )

2025-11-13 19:17:27 +08:00

wfp8afp8_sparse_gemm

【New Feature】支持Fp8 group Gemm 24稀疏 (#3463 )

2025-08-19 02:54:47 -07:00

append_attention.cu

[Others]get_block_shape_and_split_kv_block clean code (#5123 )

2025-11-20 16:40:04 +08:00

beam_search_softmax.cu

[LLM] First commit the llm deployment code

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

cpp_extensions.cc

[Feature] Support noaux for eplb (#5143 )

2025-11-21 14:10:32 +08:00

cuda_multiprocess.h

[LLM] First commit the llm deployment code

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

dequant_int8.cu

[LLM] First commit the llm deployment code

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

enforce_generation.cu

[LLM] First commit the llm deployment code

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

env.h

【Inference Optimize】DeepSeek-V3-model MLA Optimize (#3886 )

2025-09-11 10:46:09 +08:00

fused_get_rotary_embedding.cu

rename fused_get_rope.cu (#3752 )

2025-09-03 10:54:34 +08:00

fused_hadamard_quant_fp8.cu

[Optimize] Optimize tensorwise fp8 performance (#2729 )

2025-07-07 20:06:28 +08:00

fused_neox_rope_embedding.cu

[Feature] Optim PaddleOCR-VL (#4873 )

2025-11-07 14:56:44 +08:00

fused_rotary_position_encoding.cu

[Feature] DeepseekV3 use pd_build_static_op (#2948 )

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

gather_idx.cu

[LLM] First commit the llm deployment code

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

gelu_tanh.cu

[Feature] Optim PaddleOCR-VL (#4873 )

2025-11-07 14:56:44 +08:00

get_data_ptr_ipc.cu

Sync v2.0 version of code to github repo

2025-06-29 23:29:37 +00:00

get_img_boundaries.cc

enhance set_stop_value_multi_ends and standardize the registration of some operators (#4525 )

2025-10-21 22:06:06 +08:00

get_mm_split_fuse.cc

enhance set_stop_value_multi_ends and standardize the registration of some operators (#4525 )

2025-10-21 22:06:06 +08:00

get_output_ep.cc

[OPs] Universal optimization and Fix early_stop cuda 700 (#3375 )

2025-08-14 22:40:44 +08:00

get_output_msg_with_topk.cc

ep support logprob (#4089 ) (#4151 )

2025-09-19 14:07:31 +08:00

get_output.cc

[LLM] First commit the llm deployment code

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

get_padding_offset_system.cu

[LLM] First commit the llm deployment code

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

get_padding_offset.cu

[get_padding_offset.] clean get_padding_offset.cu (#4777 )

2025-11-05 10:47:40 +08:00

get_position_ids_and_mask_encoder_batch.cu

[Feature] DeepseekV3 use pd_build_static_op (#2948 )

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

helper.cu

[Feature][OP] Append Attn Support CUDA-PDL (#5072 )

2025-11-17 20:47:33 +08:00

helper.h

check METAX_GPU (#5114 )

2025-11-19 16:02:21 +08:00

init_signal_layerwise.cc

[LLM] First commit the llm deployment code

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

ipc_sent_key_value_cache_by_remote_ptr.cu

[BugFix]remove redundant includes (#4312 )

2025-09-30 17:54:19 +08:00

limit_thinking_content_length_v1.cu

[Metax] optimize flash mla (#4915 )

2025-11-12 16:43:46 +08:00

limit_thinking_content_length_v2.cu

[Metax] optimize flash mla (#4915 )

2025-11-12 16:43:46 +08:00

merge_prefill_decode_output.cu

集中式支持fa3 (#3112 )

2025-08-01 18:03:36 +08:00

msg_utils.h

polish code with new pre-commit rule (#2923 )

2025-07-19 23:19:27 +08:00

multi_head_latent_attention.cu

【Hackathon 9th No.86】autogen MoeFastHardamardImplWrapper template_instantiation (#4592 )

2025-10-30 10:28:36 +08:00

ngram_mask.cu

[LLM] First commit the llm deployment code

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

noaux_tc_redundant.cu

[Feature] Support noaux for eplb (#5143 )

2025-11-21 14:10:32 +08:00

noaux_tc.cu

Fix noaux_tc cuda Error 700 in CUDAGraph (#4174 )

2025-09-23 18:41:33 +08:00

noauxtc_kernel.h

[Fix] fix eplb noaux (#5239 )

2025-11-26 17:50:51 +08:00

open_shm_and_get_meta_signal.cc

Sync v2.0 version of code to github repo

2025-06-29 23:29:37 +00:00

per_token_quant_fp8.cu

[OP] Add InferShape&InferDtype for per_token_quant_padding (#4667 )

2025-10-30 10:28:26 +08:00

read_data_ipc.cu

[LLM] First commit the llm deployment code

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

read_ids.py

polish code with new pre-commit rule (#2923 )

2025-07-19 23:19:27 +08:00

read_temp_ids.py

polish code with new pre-commit rule (#2923 )

2025-07-19 23:19:27 +08:00

rebuild_padding.cu

[Feature] support mtp logprob (#4464 )

2025-10-20 15:18:12 +08:00

recover_decode_task.cu

support mtp in v1_scheduler mode (#3695 )

2025-09-04 17:39:59 +08:00

remote_cache_kv_ipc.cc

polish code with new pre-commit rule (#2923 )

2025-07-19 23:19:27 +08:00

remote_cache_kv_ipc.h

[BugFix] [PD Disaggregation] fix v1 scheduler prefill node profile run & ipc transfer protocol (#5132 )

2025-11-20 21:39:22 +08:00

save_output_msg_with_topk.cc

fix typos (#3684 )

2025-09-01 17:50:17 +08:00

save_with_output_msg.cc

fix typos (#3684 )

2025-09-01 17:50:17 +08:00

save_with_output_msg.h

[LLM] First commit the llm deployment code

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

save_with_output.cc

[LLM] First commit the llm deployment code

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

scaled_gemm_f8_i4_f16_gemm.cu

[LLM] First commit the llm deployment code

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

scaled_gemm_f8_i4_f16_weight_quantize.cu

fix scaled_gemm_f8_i4_f16_weight_quantize input (#3685 )

2025-08-29 11:04:04 +08:00

seqs2seqs.cu

[LLM] First commit the llm deployment code

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

set_data_ipc.cu

dcu adapter ernie45t (#2756 )

2025-07-09 18:56:27 +08:00

set_flags.cu

[LLM] First commit the llm deployment code

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

set_mask_value.cu

[LLM] First commit the llm deployment code

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

set_value_by_flags_and_idx.cu

Optimizing the performance of think length limit using custom operators (#4279 )

2025-10-20 21:09:13 +08:00

share_external_data.cu

polish code with new pre-commit rule (#2923 )

2025-07-19 23:19:27 +08:00

step_reschedule.cu

fix typos (#3684 )

2025-09-01 17:50:17 +08:00

step_system_cache.cu

polish code with new pre-commit rule (#2923 )

2025-07-19 23:19:27 +08:00

step.cu

Adapt for iluvatar gpu (#2684 )

2025-07-07 16:53:14 +08:00

stop_generation_multi_ends.cu

enhance set_stop_value_multi_ends and standardize the registration of some operators (#4525 )

2025-10-21 22:06:06 +08:00

stop_generation.cu

[LLM] First commit the llm deployment code

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

swap_cache_batch.cu

[Feature] dyc8 support prefixcache (#5125 )

2025-11-21 19:46:26 +08:00

swap_cache.cu

polish code with new pre-commit rule (#2923 )

2025-07-19 23:19:27 +08:00

system2group.cu

[LLM] First commit the llm deployment code

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

text_image_gather_scatter.cu

[Graph Optimization] Fix IR graph dependency error exposed after enabling SOT by updating the return value of TextImageGatherScatter (#4610 )

2025-10-28 18:31:23 +08:00

text_image_index_out.cu

[Feat] ernie4_5_vl_moe support CudaGraph (#3226 )

2025-09-10 13:11:57 +08:00

token_penalty_multi_scores.cu

[Feature] support bad_words (#3055 )

2025-07-30 09:31:29 +08:00

token_penalty_only_once.cu

[LLM] First commit the llm deployment code

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

token_transfer.hpp

fix typos (#3684 )

2025-09-01 17:50:17 +08:00

transfer_output.cc

[LLM] First commit the llm deployment code

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

tune_cublaslt_gemm.cu

polish code with new pre-commit rule (#2923 )

2025-07-19 23:19:27 +08:00

unset_data_ipc.cu

[feat] support prefix cache clearing when /clear_load_weight is called (#4008 )

2025-09-28 19:42:53 +08:00

update_attn_mask_offsets.cu

[BugFix]Fix attention mask bug in D-Node of PD-split mode (#5245 )

2025-11-26 17:56:28 +08:00

update_inputs_beam.cu

[XPU] bind some OPs for VL model with pybind (#4522 )

2025-10-27 10:50:08 +08:00

update_inputs_v1.cu

[XPU] bind some OPs for VL model with pybind (#4522 )

2025-10-27 10:50:08 +08:00

update_inputs.cu

[XPU] bind some OPs for VL model with pybind (#4522 )

2025-10-27 10:50:08 +08:00

update_split_fuse_input.cu

fix typos (#3684 )

2025-09-01 17:50:17 +08:00