Skip to content

Commit e592bb0

Browse files
Trigger only on supported HW
1 parent 072f3d8 commit e592bb0

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tests/test-quantize-fns.cpp

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -179,12 +179,18 @@ int main(int argc, char * argv[]) {
179179
// Test i8mm path (nrc=2) for supported types
180180
if (type == GGML_TYPE_Q4_0 || type == GGML_TYPE_Q4_1 || type == GGML_TYPE_Q8_0 ||
181181
type == GGML_TYPE_Q4_K || type == GGML_TYPE_Q6_K) {
182+
#if defined(__ARM_FEATURE_MATMUL_INT8)
182183
const float vec_dot_error_i8mm = dot_product_error(qfns, qfns_cpu, test_size, test_data.data(), test_data2.data(), 2);
183184
failed = !(vec_dot_error_i8mm < max_allowed_error);
184185
num_failed += failed;
185186
if (failed || verbose) {
186187
printf("%5s dot product error (i8mm): %s (%f)\n", ggml_type_name(type), RESULT_STR[failed], vec_dot_error_i8mm);
187188
}
189+
#else
190+
if (verbose) {
191+
printf("%5s dot product (i8mm): SKIPPED (not supported)\n", ggml_type_name(type));
192+
}
193+
#endif
188194
}
189195
}
190196
}

0 commit comments

Comments
 (0)