Chunyuan WU 6005eceee3 [CPU] remove process_group from inputs of shm_allreduce and shm_allgather (#7486)

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

..

activation.cpp

…

bmm.cpp

…

CMakeLists.txt

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

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

common.h

…

decode.cpp

…

extend.cpp

…

gemm_fp8.cpp

…

gemm_int8.cpp

…

gemm.cpp

[CPU] [BF16] Call fused_experts_cpu, weight_packed_linear and bmm_cpu kernel in DeepSeek model (#6641)

2025-06-25 01:43:33 -07:00

gemm.h

…

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

…

moe_int8.cpp

…

moe.cpp

…

norm.cpp

…

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

…

rope.cpp

…

shm.cpp

…

shm.h

…

topk.cpp

…

torch_extension_cpu.cpp

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

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

vec.h

…