co63oc b6edd15d55 fix scaled_gemm_f8_i4_f16_weight_quantize input (#3685)

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

..

append_attn

support fa3 rope3d (#3622)

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

common

…

custom_all_reduce

Add custom op declaration for all_reduce (#3473)

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

cutlass_extensions

[Optimize]support machete weight only gemm (#3561)

2025-08-28 09:49:58 +08:00

cutlass_kernels

…

flash_mask_attn

[New Feature] fa3 支持flash mask (#3184)

2025-08-05 12:20:48 +08:00

fp8_gemm_with_cutlass

…

glog

…

int8_gemm_with_cutlass

…

machete

[Optimize]support machete weight only gemm (#3561)

2025-08-28 09:49:58 +08:00

mla_attn

…

moe

【New Feature】集中式支持w4afp8 (#3644)

2025-08-28 10:53:24 +08:00

quantization

…

sample_kernels

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

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

speculate_decoding

[Feature][MTP]support new speculative decoding method named hybrid mtp with ngram (#3610)

2025-08-26 14:29:22 +08:00

w4afp8_gemm

【New Feature】集中式支持w4afp8 (#3644)

2025-08-28 10:53:24 +08:00

wfp8afp8_sparse_gemm

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

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

append_attention.cu

Add with_output version AppendAttention (#3302)

2025-08-28 17:10:18 +08:00

beam_search_softmax.cu

…

cpp_extensions.cc

Add with_output version AppendAttention (#3302)

2025-08-28 17:10:18 +08:00

cuda_multiprocess.h

…

dequant_int8.cu

…

enforce_generation.cu

…

env.h

…

extract_text_token_output.cu

…

fused_get_rope.cu

…

fused_hadamard_quant_fp8.cu

…

fused_rotary_position_encoding.cu

…

gather_idx.cu

…

get_data_ptr_ipc.cu

…

get_img_boundaries.cc

…

get_mm_split_fuse.cc

…

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

…

get_output.cc

…

get_padding_offset_system.cu

…

get_padding_offset.cu

enable dcu ci (#3402)

2025-08-29 10:23:08 +08:00

get_position_ids_and_mask_encoder_batch.cu

…

helper.cu

…

helper.h

support w4afp8 EP inference (#3044)

2025-08-25 11:27:45 +08:00

init_signal_layerwise.cc

…

ipc_sent_key_value_cache_by_remote_ptr.cu

…

merge_prefill_decode_output.cu

集中式支持fa3 (#3112)

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

msg_utils.h

…

multi_head_latent_attention.cu

…

ngram_mask.cu

…

noaux_tc.cu

[Bugfix] Fix model accuracy in some ops (#3231)

2025-08-08 17:30:37 +08:00

noauxtc_kernel.h

revise get_moe_scores (#3164)

2025-08-05 16:43:07 +08:00

open_shm_and_get_meta_signal.cc

…

per_token_quant_fp8.cu

Support 45t fp8 8 GPU (#3659)

2025-08-28 10:52:53 +08:00

read_data_ipc.cu

…

read_ids.py

…

read_temp_ids.py

…

rebuild_padding.cu

[Code Simplification] remove cum_offsets (#3410)

2025-08-18 20:21:25 +08:00

recover_decode_task.cu

…

remote_cache_kv_ipc.cc

…

remote_cache_kv_ipc.h

…

save_output_msg_with_topk.cc

…

save_with_output_msg.cc

Supports DP+TP+EP hybrid parallel deployment strategy (#3489)

2025-08-26 00:04:01 -07: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

…

set_flags.cu

…

set_mask_value.cu

…

set_value_by_flags.cu

…

share_external_data.cu

…

step_reschedule.cu

…

step_system_cache.cu

…

step.cu

…

stop_generation_multi_ends.cu

…

stop_generation.cu

…

swap_cache_batch.cu

…

swap_cache.cu

…

system2group.cu

…

text_image_gather_scatter.cu

…

text_image_index_out.cu

…

token_penalty_multi_scores.cu

…

token_penalty_only_once.cu

…

token_transfer.hpp

…

transfer_output.cc

…

tune_cublaslt_gemm.cu

…

update_inputs_beam.cu

…

update_inputs_v1.cu

…

update_inputs.cu

…

update_split_fuse_input.cu

…