This change upstreams llamafile's cpu matrix multiplication kernels for ppc64le using MMA builtins for quantised int8 datatype. This change results in 10% - 70% improvement in total speed(ie all tokens/total time), across various batch sizes. The patch is tested with Meta-Lllama-3-8B, Mistral-7B, Llama-2-7B-chat-hf models on a IBM POWER10 machine. Signed-off-by: Amrita H S <amritahs@linux.vnet.ibm.com> |
||
|---|---|---|
| .. | ||
| amx | ||
| cmake | ||
| llamafile | ||
| CMakeLists.txt | ||
| cpu-feats-x86.cpp | ||
| ggml-cpu-aarch64.cpp | ||
| ggml-cpu-aarch64.h | ||
| ggml-cpu-hbm.cpp | ||
| ggml-cpu-hbm.h | ||
| ggml-cpu-impl.h | ||
| ggml-cpu-quants.c | ||
| ggml-cpu-quants.h | ||
| ggml-cpu-traits.cpp | ||
| ggml-cpu-traits.h | ||
| ggml-cpu.c | ||
| ggml-cpu.cpp | ||