Zaili Wang d6bd2d1126 [CPU] layernorm & fused add-layernorm kernels (#14074)

2025-12-11 16:58:23 -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

…

bmm.cpp

…

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

…

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

…

gemm.cpp

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

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

gemm.h

…

interface.cpp

…

moe_fp8.cpp

…

moe_int8.cpp

…

moe.cpp

…

norm.cpp

[CPU] layernorm & fused add-layernorm kernels (#14074)

2025-12-11 16:58:23 -08:00

numa_utils.cpp

…

qkv_proj.cpp

…

rope.cpp

…

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

…

torch_extension_cpu.cpp

[CPU] layernorm & fused add-layernorm kernels (#14074)

2025-12-11 16:58: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