huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 10:26:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add the remaining quantized tests to the wasm suite. (#980)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-28 08:42:56 +01:00
committed by GitHub
parent 9cb110c44c
commit 8a49e01b9d
1 changed files with 32 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
candle-wasm-tests/tests/quantized_tests.rs
View File

@ -134,8 +134,8 @@ fn quantized_matmul_q40() -> Result<()> {
} }
#[wasm_bindgen_test] #[wasm_bindgen_test]
fn quantized_matmul_q4k() -> Result<()> { fn quantized_matmul_q50() -> Result<()> {
ggml_matmul_error_test::<candle::quantized::k_quants::BlockQ4K>()?; ggml_matmul_error_test::<candle::quantized::k_quants::BlockQ5_0>()?;
Ok(()) Ok(())
} }
@ -144,3 +144,33 @@ fn quantized_matmul_q80() -> Result<()> {
ggml_matmul_error_test::<candle::quantized::k_quants::BlockQ8_0>()?; ggml_matmul_error_test::<candle::quantized::k_quants::BlockQ8_0>()?;
Ok(()) 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(())
}