diff --git a/candle-core/tests/quantized_tests.rs b/candle-core/tests/quantized_tests.rs index 84223ed4e..a7811ca5a 100644 --- a/candle-core/tests/quantized_tests.rs +++ b/candle-core/tests/quantized_tests.rs @@ -47,6 +47,10 @@ fn test_matmul( } fn quantized_matmul(device: &Device) -> Result<()> { + // TODO Enable this later when we enable cuda. + if device.is_cuda() { + return Ok(()); + } let (m, k, n) = (3, 64, 4); let lhs = (0..(m * k)).map(|v| v as f32).collect::>(); let tensor_lhs = Tensor::from_slice(&lhs, (m, k), device)?; @@ -101,6 +105,10 @@ fn quantized_matmul(device: &Device) -> Result<()> { } fn quantized_matmul_neg(device: &Device) -> Result<()> { + // TODO Enable this later when we enable cuda. + if device.is_cuda() { + return Ok(()); + } let (m, k, n) = (3, 64, 4); let lhs = (0..(m * k)) .map(|v| v as f32 - (m * k) as f32 / 2.0) @@ -170,6 +178,10 @@ test_device!( ); fn quantize_q4_0(device: &Device) -> Result<()> { + // TODO Enable this later when we enable cuda. + if device.is_cuda() { + return Ok(()); + } let src = (0..32 * 4).map(|v| v as f32).collect::>(); let src = Tensor::from_slice(&src, (32 * 4,), device)?; @@ -197,6 +209,10 @@ fn quantize_q4_0(device: &Device) -> Result<()> { } fn quantize_q4_1(device: &Device) -> Result<()> { + // TODO Enable this later when we enable cuda. + if device.is_cuda() { + return Ok(()); + } let src = (0..32 * 4).map(|v| v as f32).collect::>(); let src = Tensor::from_slice(&src, (32 * 4,), device)?; let quant = quantized::QTensor::quantize(&src, GgmlDType::Q4_1)?; @@ -223,6 +239,10 @@ fn quantize_q4_1(device: &Device) -> Result<()> { } fn quantize_q5_0(device: &Device) -> Result<()> { + // TODO Enable this later when we enable cuda. + if device.is_cuda() { + return Ok(()); + } let src = (0..32 * 4).map(|v| v as f32).collect::>(); let src = Tensor::from_slice(&src, (32 * 4,), device)?; let quant = quantized::QTensor::quantize(&src, GgmlDType::Q5_0)?; @@ -249,6 +269,10 @@ fn quantize_q5_0(device: &Device) -> Result<()> { } fn quantize_q5_1(device: &Device) -> Result<()> { + // TODO Enable this later when we enable cuda. + if device.is_cuda() { + return Ok(()); + } let src = (0..32 * 4).map(|v| v as f32).collect::>(); let src = Tensor::from_slice(&src, (32 * 4,), device)?; let quant = quantized::QTensor::quantize(&src, GgmlDType::Q5_1)?; @@ -349,6 +373,10 @@ fn ggml_quantization_error_test(dtype: GgmlDType, device: &Device, max_error: f3 } fn quantize_q2k(device: &Device) -> Result<()> { + // TODO Enable this later when we enable cuda. + if device.is_cuda() { + return Ok(()); + } let dtype = GgmlDType::Q2K; let src = get_test_vector2(0.5, 1024, device)?; @@ -383,6 +411,10 @@ fn quantize_q2k(device: &Device) -> Result<()> { } fn quantize_q3k(device: &Device) -> Result<()> { + // TODO Enable this later when we enable cuda. + if device.is_cuda() { + return Ok(()); + } let dtype = GgmlDType::Q3K; let src = get_test_vector2(0.5, 1024, device)?; let quant = quantized::QTensor::quantize(&src, dtype)?; @@ -416,6 +448,10 @@ fn quantize_q3k(device: &Device) -> Result<()> { } fn quantize_q4k(device: &Device) -> Result<()> { + // TODO Enable this later when we enable cuda. + if device.is_cuda() { + return Ok(()); + } let dtype = GgmlDType::Q4K; let src = get_test_vector2(0.5, 1024, device)?; let quant = quantized::QTensor::quantize(&src, dtype)?; @@ -449,6 +485,10 @@ fn quantize_q4k(device: &Device) -> Result<()> { } fn quantize_q5k(device: &Device) -> Result<()> { + // TODO Enable this later when we enable cuda. + if device.is_cuda() { + return Ok(()); + } let dtype = GgmlDType::Q5K; let src = get_test_vector2(0.5, 1024, device)?; let quant = quantized::QTensor::quantize(&src, dtype)?; @@ -482,6 +522,10 @@ fn quantize_q5k(device: &Device) -> Result<()> { } fn quantize_q6k(device: &Device) -> Result<()> { + // TODO Enable this later when we enable cuda. + if device.is_cuda() { + return Ok(()); + } let dtype = GgmlDType::Q6K; let src = get_test_vector2(0.5, 1024, device)?; let quant = quantized::QTensor::quantize(&src, dtype)?; @@ -515,6 +559,10 @@ fn quantize_q6k(device: &Device) -> Result<()> { } fn quantize_q8k(device: &Device) -> Result<()> { + // TODO Enable this later when we enable cuda. + if device.is_cuda() { + return Ok(()); + } let dtype = GgmlDType::Q8K; let src = get_test_vector2(0.5, 1024, device)?; let quant = quantized::QTensor::quantize(&src, dtype)?;