blzheng d257bf87b9 [CPU] add mamba fla kernels for Qwen3-next (#12324)

2025-12-06 14:16:23 +08:00

..

mamba

[CPU] add mamba fla kernels for Qwen3-next (#12324)

2025-12-06 14:16:23 +08:00

model

[CPU] add fused_qkvzba_split_reshape_cat kernel for Qwen3-next (#12330)

2025-12-03 23:46:08 +08:00

activation.cpp

[CPU] Add gelu_and_mul kernel in sgl-kernel and add ut (#9300)

2025-09-08 23:23:13 -07:00

bmm.cpp

[CPU] Fix build issue (#6419)

2025-05-21 11:17:10 -07:00

CMakeLists.txt

[CPU] Support chunk_gated_delta_rule kernel for Qwen3-Next (#12441)

2025-12-03 17:03:48 +08:00

common.h

[CPU] add mamba fla kernels for Qwen3-next (#12324)

2025-12-06 14:16:23 +08:00

decode.cpp

[CPU] Optimize FP16 decode_attention_cpu (#10652)

2025-10-22 21:39:51 -07:00

extend.cpp

[CPU] Support chunk_gated_delta_rule kernel for Qwen3-Next (#12441)

2025-12-03 17:03:48 +08:00

gemm_fp8.cpp

use faster covnersion from float8_e4m3fn to bfloat16 (#12316)

2025-12-04 12:34:05 +08:00

gemm_int8.cpp

Optimize prefill performance on cpu backend (#8750)

2025-08-28 17:21:55 -07:00

gemm.cpp

[CPU] Optimize small oc GEMM for Qwen3-next on CPU (#12446)

2025-12-04 00:38:47 -08:00

gemm.h

Optimize prefill performance on cpu backend (#8750)

2025-08-28 17:21:55 -07:00

interface.cpp

[CPU] remove process_group from inputs of shm_allreduce and shm_allgather (#7486)

2025-06-30 21:54:11 -07:00

moe_fp8.cpp

Optimize prefill performance on cpu backend (#8750)

2025-08-28 17:21:55 -07:00

moe_int8.cpp

Optimize prefill performance on cpu backend (#8750)

2025-08-28 17:21:55 -07:00

moe.cpp

Optimize prefill performance on cpu backend (#8750)

2025-08-28 17:21:55 -07:00

norm.cpp

Add fused_rmsnorm_gated_cpu kernel for CPU to support Qwen3-Next (#11577)

2025-11-21 01:33:31 +08:00

numa_utils.cpp

[CPU] add c++ kernel to bind CPU cores and memory node (#7524)

2025-06-29 19:45:25 -07:00

qkv_proj.cpp

Optimize prefill performance on cpu backend (#8750)

2025-08-28 17:21:55 -07:00

rope.cpp

Add CPU optimized kernels for topk and rope fusions (#6456)

2025-06-02 17:37:34 -07:00

shm.cpp

CPU: map changes from developing branch in sgl-kernel (#6833)

2025-06-10 01:08:15 -07:00

shm.h

CPU: map changes from developing branch in sgl-kernel (#6833)

2025-06-10 01:08:15 -07:00

topk.cpp

[CPU] Fix TP padding case with weight block size (#8243)

2025-11-07 03:24:48 +08:00

torch_extension_cpu.cpp

[CPU] add mamba fla kernels for Qwen3-next (#12324)

2025-12-06 14:16:23 +08:00

vec_pack.h

[CPU] Support chunk_gated_delta_rule kernel for Qwen3-Next (#12441)

2025-12-03 17:03:48 +08:00

vec.h

use faster covnersion from float8_e4m3fn to bfloat16 (#12316)

2025-12-04 12:34:05 +08:00