Skip to content

Commit 87e5d92

Browse files
authored
Merge pull request #641 from oreomaker/dev
feat: x86 matmul and transpose support (SDPA llama runnable)
2 parents d257c50 + 60962ea commit 87e5d92

9 files changed

Lines changed: 1379 additions & 34 deletions

File tree

mllm/backends/cpu/kernels/Kernels.hpp

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -8,12 +8,14 @@
88
#include "mllm/utils/CPUArchHelper.hpp"
99

1010
#if defined(MLLM_HOST_ARCH_X86_64) || defined(MLLM_HOST_ARCH_X86)
11-
#include "mllm/backends/cpu/kernels/x86/fill.hpp" // IWYU pragma: export
12-
#include "mllm/backends/cpu/kernels/x86/silu.hpp" // IWYU pragma: export
13-
#include "mllm/backends/cpu/kernels/x86/sigmoid.hpp" // IWYU pragma: export
14-
#include "mllm/backends/cpu/kernels/x86/softmax.hpp" // IWYU pragma: export
15-
#include "mllm/backends/cpu/kernels/x86/rmsnorm.hpp" // IWYU pragma: export
16-
#include "mllm/backends/cpu/kernels/x86/gelu.hpp" // IWYU pragma: export
11+
#include "mllm/backends/cpu/kernels/x86/fill.hpp" // IWYU pragma: export
12+
#include "mllm/backends/cpu/kernels/x86/silu.hpp" // IWYU pragma: export
13+
#include "mllm/backends/cpu/kernels/x86/sigmoid.hpp" // IWYU pragma: export
14+
#include "mllm/backends/cpu/kernels/x86/softmax.hpp" // IWYU pragma: export
15+
#include "mllm/backends/cpu/kernels/x86/rmsnorm.hpp" // IWYU pragma: export
16+
#include "mllm/backends/cpu/kernels/x86/gelu.hpp" // IWYU pragma: export
17+
#include "mllm/backends/cpu/kernels/x86/transpose.hpp" // IWYU pragma: export
18+
#include "mllm/backends/cpu/kernels/x86/mllm_blas/mllm_blas_sgemm.hpp" // IWYU pragma: export
1719
#endif
1820

1921
#if defined(MLLM_HOST_ARCH_ARM64) || defined(MLLM_HOST_ARCH_ARM)

0 commit comments

Comments
 (0)