From 8a49e01b9d9e13732d83f79afa2f850e2ba7fdae Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Thu, 28 Sep 2023 08:42:56 +0100
Subject: [PATCH] Add the remaining quantized tests to the wasm suite. (#980)

---
 candle-wasm-tests/tests/quantized_tests.rs | 34 ++++++++++++++++++++--
 1 file changed, 32 insertions(+), 2 deletions(-)

diff --git a/candle-wasm-tests/tests/quantized_tests.rs b/candle-wasm-tests/tests/quantized_tests.rs
index d16cf497..643eb2f0 100644
--- a/candle-wasm-tests/tests/quantized_tests.rs
+++ b/candle-wasm-tests/tests/quantized_tests.rs
@@ -134,8 +134,8 @@ fn quantized_matmul_q40() -> Result<()> {
 }
 
 #[wasm_bindgen_test]
-fn quantized_matmul_q4k() -> Result<()> {
-    ggml_matmul_error_test::<candle::quantized::k_quants::BlockQ4K>()?;
+fn quantized_matmul_q50() -> Result<()> {
+    ggml_matmul_error_test::<candle::quantized::k_quants::BlockQ5_0>()?;
     Ok(())
 }
 
@@ -144,3 +144,33 @@ fn quantized_matmul_q80() -> Result<()> {
     ggml_matmul_error_test::<candle::quantized::k_quants::BlockQ8_0>()?;
     Ok(())
 }
+
+#[wasm_bindgen_test]
+fn quantized_matmul_q2k() -> Result<()> {
+    ggml_matmul_error_test::<candle::quantized::k_quants::BlockQ2K>()?;
+    Ok(())
+}
+
+#[wasm_bindgen_test]
+fn quantized_matmul_q3k() -> Result<()> {
+    ggml_matmul_error_test::<candle::quantized::k_quants::BlockQ3K>()?;
+    Ok(())
+}
+
+#[wasm_bindgen_test]
+fn quantized_matmul_q4k() -> Result<()> {
+    ggml_matmul_error_test::<candle::quantized::k_quants::BlockQ4K>()?;
+    Ok(())
+}
+
+#[wasm_bindgen_test]
+fn quantized_matmul_q5k() -> Result<()> {
+    ggml_matmul_error_test::<candle::quantized::k_quants::BlockQ5K>()?;
+    Ok(())
+}
+
+#[wasm_bindgen_test]
+fn quantized_matmul_q6k() -> Result<()> {
+    ggml_matmul_error_test::<candle::quantized::k_quants::BlockQ6K>()?;
+    Ok(())
+}