Cao E 274581fb77 Add support for more batch sizes in cpu_graph_runner (#13881)

2026-03-19 09:50:56 -07:00

..

aarch64

…

mamba

…

model

…

x86_64

…

activation.cpp

…

bmm.cpp

[CPU] Add FP8 Bmm support (#9744)

2026-03-18 22:19:48 -07:00

CMakeLists.txt

…

common.h

[CPU] Optimize image preprocessor performance for Qwen2VLImageProcessorFast (#15168)

2026-03-18 22:18:15 -07:00

conv3d.cpp

optimize conv3d used in patch embedding (#16040)

2026-03-18 22:17:53 -07:00

decode.cpp

[CPU] optimize flash_attn_varlen_func (#15708)

2026-01-29 22:07:05 -08:00

extend.cpp

[CPU] optimize flash_attn_varlen_func (#15708)

2026-01-29 22:07:05 -08:00

flash_attn.cpp

[CPU] fix bug in AVX512 implementation of flash_attn_softmax (#20220)

2026-03-18 22:18:47 -07:00

flash_attn.h

[CPU] fix bug in AVX512 implementation of flash_attn_softmax (#20220)

2026-03-18 22:18:47 -07:00

gemm_fp8.cpp

[CPU] Add FP8 Bmm support (#9744)

2026-03-18 22:19:48 -07:00

gemm_int4.cpp

[CPU][INT4] Add INT4 kernels for CPU (#8226)

2026-01-29 22:30:13 -08:00

gemm_int8.cpp

…

gemm.cpp

…

gemm.h

[CPU] Add FP8 Bmm support (#9744)

2026-03-18 22:19:48 -07:00

interface.cpp

…

moe_fp8.cpp

…

moe_int4.cpp

[CPU][INT4] Add INT4 kernels for CPU (#8226)

2026-01-29 22:30:13 -08:00

moe_int8.cpp

…

moe.cpp

[CPU][INT4] Add INT4 kernels for CPU (#8226)

2026-01-29 22:30:13 -08:00

norm.cpp

[CPU] support LayerNorm with 3D shape (#15075)

2026-03-18 22:15:24 -07:00

numa_utils.cpp

[CPU] improve numa memory binding (#19666)

2026-03-18 22:15:50 -07:00

preprocessor.cpp

[CPU] Optimize image preprocessor performance for Qwen2VLImageProcessorFast (#15168)

2026-03-18 22:18:15 -07:00

qkv_proj.cpp

[CPU] Add FP8 Bmm support (#9744)

2026-03-18 22:19:48 -07:00

rope.cpp

[CPU] Add mrope kernel for Qwen3-vl (#12531)

2026-03-18 22:12:48 -07:00

shm.cpp

…

shm.h

…

topk.cpp

[CPU] Optimize Qwen3-next model on CPU (#12525)

2026-01-29 22:03:58 -08:00

torch_extension_cpu.cpp

Add support for more batch sizes in cpu_graph_runner (#13881)

2026-03-19 09:50:56 -07:00

vec_pack.h

[CPU] optimize flash_attn_varlen_func (#15708)

2026-01-29 22:07:05 -08:00

vec.h

[CPU] optimize flash_attn_varlen_func (#15708)

2026-01-29 22:07:05 -08:00