fxyfxy777 2ada119a38 [Optimize] optimize mask_quant & swiglu (#6222) ...

* optimize mask_quant op speed up 1.5

* fix calculate sequence

* add fused

* rm log

* push kernel code

* add ut

* accuracy ok

* add ue8m0

* add ut

* add merge develop

* rm ut of mask_per_token_quant

2026-02-02 13:52:38 +08:00

..

append_attn

[Model Runner] Prepare token count and move FA3 initialization into the graph (#6170)

2026-01-26 12:16:57 +08:00

common

…

custom_all_reduce

Support setting communication groups in custom_allreduce and the all-to-all\transpose fused operator during the decoding phase. (#5917)

2026-01-12 14:09:39 +08:00

cutlass_extensions

fix typos (#3684)

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

cutlass_kernels

[Feature] Support fusedmoe on Blackwell (#5325)

2025-12-16 11:58:50 +08:00

flash_mask_attn

remove unneeded para from flash_mask_attention (#6218)

2026-01-27 14:04:27 +08:00

fp8_gemm_with_cutlass

…

glog

…

int8_gemm_with_cutlass

fix typos (#3684)

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

machete

[BugFix] Fix wint4 ep issue caused by empty run (#5870)

2026-01-05 14:24:37 +08:00

mla_attn

[BugFix] fix BatchMLAWithPagedKVCacheKernel O_tmp (#5895)

2026-01-06 15:39:06 +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

Revert "[Feature] Support Ernie FP8 on sm100 (#5593)" (#6275)

2026-01-30 11:22:01 +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

[Metax] refactor cutlass moe and optimize flash attention (#5361)

2025-12-10 17:15:17 +08:00

speculate_decoding

[Metax] adapt to the latest develop (#6282)

2026-01-29 23:21:20 -08:00

w4afp8_gemm

[BugFix] fix wint2 (#6109)

2026-01-20 21:46:21 +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

…

cpp_extensions.cc

[Optimize] optimize mask_quant & swiglu (#6222)

2026-02-02 13:52:38 +08:00

cuda_multiprocess.h

…

dequant_int8.cu

…

enforce_generation.cu

…

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

…

fused_mask_swiglu_fp8_quant_kernel.cu

[Optimize] optimize mask_quant & swiglu (#6222)

2026-02-02 13:52:38 +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

…

gather_idx.cu

…

gelu_tanh.cu

[Metax] adapt to the latest develop (#6282)

2026-01-29 23:21:20 -08:00

get_data_ptr_ipc.cu

…

get_img_boundaries.cc

[Models] Add Qwen3-VL Model Support (#5763)

2025-12-29 17:39:33 +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

[PD Disaggregation] Distinguish the pipelines for sending kv signal in different prefill (#5514)

2025-12-12 14:05:36 +08:00

get_output_msg_with_topk.cc

ep support logprob (#4089) (#4151)

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

get_output.cc

[PD Disaggregation] support DP via v1 router and decouple DP and EP (#5197)

2025-12-04 15:38:43 +08:00

get_padding_offset_system.cu

…

get_padding_offset.cu

[Model Runner] Prepare token count and move FA3 initialization into the graph (#6170)

2026-01-26 12:16:57 +08:00

get_position_ids_and_mask_encoder_batch.cu

…

helper.cu

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

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

helper.h

remove unneeded para from flash_mask_attention (#6218)

2026-01-27 14:04:27 +08:00

init_signal_layerwise.cc

…

ipc_sent_key_value_cache_by_remote_ptr.cu

[Feature] support rdma pd dy-c8 (#5788)

2026-01-07 14:55:25 +08:00

limit_thinking_content_length_v1.cu

[BugFix] Fix limit_thinking early return logic in CUDA kernels (#5471)

2025-12-10 11:03:19 +08:00

limit_thinking_content_length_v2.cu

[BugFix] Fix limit_thinking early return logic in CUDA kernels (#5471)

2025-12-10 11:03:19 +08:00

merge_prefill_decode_output.cu

集中式支持fa3 (#3112)

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

msg_utils.h

…

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

…

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

[Others] clean code (#5691)

2025-12-24 11:28:47 +08:00

open_shm_and_get_meta_signal.cc

…

read_data_ipc.cu

…

read_ids.py

…

read_temp_ids.py

…

reasoning_phase_token_constraint.cu

[Others] Support constrained decoding when enable_thinking is false (#6248)

2026-01-28 00:05:17 -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

…

remote_cache_kv_ipc.h

[PD Disaggregation] Distinguish the pipelines for sending kv signal in different prefill (#5514)

2025-12-12 14:05:36 +08:00

save_output_msg_with_topk.cc

[Optim] Robust sync status when preempted happens (#5796)

2026-01-14 12:07:33 +08:00

save_with_output_msg.cc

[Optim] Robust sync status when preempted happens (#5796)

2026-01-14 12:07:33 +08:00

save_with_output_msg.h

…

save_with_output.cc

…

scaled_gemm_f8_i4_f16_gemm.cu

…

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

…

set_data_ipc.cu

[BugFix] fix shm opened but not closed in set_data_ipc (#5826)

2025-12-29 23:35:07 +08:00

set_flags.cu

…

set_mask_value.cu

…

set_stop.cu

[Model Runner] Refactor execute_model for GPU async scheduling (#6176)

2026-01-28 14:19:33 +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

…

step_reschedule.cu

fix typos (#3684)

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

step_system_cache.cu

…

step.cu

…

stop_generation_multi_ends.cu

[Feature] support stop_token_ids (#5399)

2025-12-09 17:49:12 +08:00

stop_generation.cu

…

swap_cache_batch.cu

[Feature] dyc8 support prefixcache (#5125)

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

swap_cache_layout.cu

[CI] Add unit test for swap_layout && remove unit test of splitwise_scheduler (#6250)

2026-01-28 19:20:20 +08:00

swap_cache.cu

…

system2group.cu

…

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

[BugFix] Fix token_penalty kernel (#6069)

2026-01-28 12:03:05 +08:00

token_penalty_only_once.cu

…

token_transfer.hpp

fix typos (#3684)

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

transfer_output.cc

…

tune_cublaslt_gemm.cu

…

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

[Model Runner] Refactor execute_model for GPU async scheduling (#6176)

2026-01-28 14:19:33 +08:00

update_inputs.cu

[Model Runner] Refactor execute_model for GPU async scheduling (#6176)

2026-01-28 14:19:33 +08:00

update_split_fuse_input.cu

fix typos (#3684)

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