blzheng e27635a02d [CPU] Add 4D input support for ROPE in sgl-kernel (#9337)

2025-12-16 17:27:39 +08:00

..

aarch64

[cpu] Implement all gather/reduce for arm64 cpu (#12527)

2025-12-08 19:03:04 +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

x86_64

[cpu] Implement all gather/reduce for arm64 cpu (#12527)

2025-12-08 19:03:04 +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

[CPU] Add Gemma3RMSNorm kernel in sgl-kernel and add ut (#9324)

2025-12-15 00:24:02 -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

[CPU] Add 4D input support for ROPE in sgl-kernel (#9337)

2025-12-16 17:27:39 +08:00

shm.cpp

[cpu] Implement all gather/reduce for arm64 cpu (#12527)

2025-12-08 19:03:04 +08:00

shm.h

[cpu] Implement all gather/reduce for arm64 cpu (#12527)

2025-12-08 19:03:04 +08: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 Gemma3RMSNorm kernel in sgl-kernel and add ut (#9324)

2025-12-15 00:24:02 -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