Skip to content

Commit a52f22f

Browse files
authored
Skip q8k and q8_1 tests on cuda (#3140)
1 parent 2bce4e5 commit a52f22f

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

candle-core/tests/quantized_tests.rs

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,9 @@ fn test_matmul(
2020
(b, m, n, k): (usize, usize, usize, usize),
2121
dtype: GgmlDType,
2222
) -> Result<()> {
23-
if device.is_metal() && (dtype == GgmlDType::Q8_1 || dtype == GgmlDType::Q8K) {
23+
if (device.is_cuda() || device.is_metal())
24+
&& (dtype == GgmlDType::Q8_1 || dtype == GgmlDType::Q8K)
25+
{
2426
return Ok(());
2527
}
2628

0 commit comments

Comments
 (0)