gemma.cpp/ops
Apoorv Reddy 48d0801fb0 Vectorize Rope for qkv dim not evenly divisible by number of lanes.
PiperOrigin-RevId: 665776602
2024-08-21 02:22:22 -07:00
..
gemma_matvec_test.cc Fix build issues when tests are enabled 2024-08-12 18:50:23 +02:00
matmul-inl.h Major MatMul update, 1.9-2.3x speedup on Zen4 via bf16 mul 2024-08-16 07:52:20 -07:00
matmul.h Major MatMul update, 1.9-2.3x speedup on Zen4 via bf16 mul 2024-08-16 07:52:20 -07:00
matmul_test.cc Major MatMul update, 1.9-2.3x speedup on Zen4 via bf16 mul 2024-08-16 07:52:20 -07:00
matvec-inl.h Split matmul into matvec; add large matrix benchmark 2024-07-30 08:29:11 -07:00
ops-inl.h Vectorize Rope for qkv dim not evenly divisible by number of lanes. 2024-08-21 02:22:22 -07:00
ops_test.cc VectorizedRopeAndMulBy. 2024-08-18 23:17:01 -07:00