diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-09-28 08:42:56 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-28 08:42:56 +0100 |
| commit | 8a49e01b9d9e13732d83f79afa2f850e2ba7fdae (patch) | |
| tree | 46076437f5cf59e7f160e46accb44e6838efcc30 /candle-wasm-tests | |
| parent | 9cb110c44c10145efefe3555ae8b9f91f4161fe2 (diff) | |
| download | candle-8a49e01b9d9e13732d83f79afa2f850e2ba7fdae.tar.gz candle-8a49e01b9d9e13732d83f79afa2f850e2ba7fdae.tar.bz2 candle-8a49e01b9d9e13732d83f79afa2f850e2ba7fdae.zip | |
Add the remaining quantized tests to the wasm suite. (#980)
Diffstat (limited to 'candle-wasm-tests')
| -rw-r--r-- | candle-wasm-tests/tests/quantized_tests.rs | 34 |
1 files 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(()) +} |