Update tests to use the set kv_group size

gkielian · May 13, 2024 · a4aed6f · a4aed6f
1 parent 154869a
commit a4aed6f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tests/test_all_activation_variations_cpu.sh b/tests/test_all_activation_variations_cpu.sh
@@ -47,6 +47,7 @@ do
     --max_iters "$max_iters" \
     --n_layer "$n_layer" \
     --n_head "$n_head" \
+    --n_kv_group "$n_kv_group" \
     --n_embd "$n_embd" \
     --eval_iters "$eval_iters" \
     --eval_interval "$eval_interval" \

diff --git a/tests/test_all_softmax_variations_cpu.sh b/tests/test_all_softmax_variations_cpu.sh
@@ -32,6 +32,7 @@ do
     --max_iters "$max_iters" \
     --n_layer "$n_layer" \
     --n_head "$n_head" \
+    --n_kv_group "$n_kv_group" \
     --n_embd "$n_embd" \
     --eval_iters "$eval_iters" \
     --eval_interval "$eval_interval" \