Yibo Cai 7871593cc8 [cpu] Implement all gather/reduce for arm64 cpu (#12527)

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

..

aarch64

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

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

mamba

…

model

…

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

…

common.h

…

decode.cpp

…

extend.cpp

…

gemm_fp8.cpp

…

gemm_int8.cpp

…

gemm.cpp

…

gemm.h

…

interface.cpp

…

moe_fp8.cpp

…

moe_int8.cpp

…

moe.cpp

…

norm.cpp

…

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

…

vec_pack.h

…

vec.h

…