Skip to content

Commit d8fd0cc

Browse files
authored
test-backend-ops : skip CPU backend by default (#6028)
1 parent b3d9786 commit d8fd0cc

File tree

1 file changed

+14
-6
lines changed

1 file changed

+14
-6
lines changed

tests/test-backend-ops.cpp

+14-6
Original file line numberDiff line numberDiff line change
@@ -2222,8 +2222,8 @@ static void usage(char ** argv) {
22222222

22232223
int main(int argc, char ** argv) {
22242224
test_mode mode = MODE_TEST;
2225-
const char * op_name = NULL;
2226-
const char * backend = NULL;
2225+
const char * op_name_filter = NULL;
2226+
const char * backend_filter = NULL;
22272227

22282228
for (int i = 1; i < argc; i++) {
22292229
if (strcmp(argv[i], "test") == 0) {
@@ -2232,14 +2232,14 @@ int main(int argc, char ** argv) {
22322232
mode = MODE_PERF;
22332233
} else if (strcmp(argv[i], "-o") == 0) {
22342234
if (i + 1 < argc) {
2235-
op_name = argv[++i];
2235+
op_name_filter = argv[++i];
22362236
} else {
22372237
usage(argv);
22382238
return 1;
22392239
}
22402240
} else if (strcmp(argv[i], "-b") == 0) {
22412241
if (i + 1 < argc) {
2242-
backend = argv[++i];
2242+
backend_filter = argv[++i];
22432243
} else {
22442244
usage(argv);
22452245
return 1;
@@ -2258,17 +2258,25 @@ int main(int argc, char ** argv) {
22582258
for (size_t i = 0; i < ggml_backend_reg_get_count(); i++) {
22592259
printf("Backend %zu/%zu (%s)\n", i + 1, ggml_backend_reg_get_count(), ggml_backend_reg_get_name(i));
22602260

2261-
if (backend != NULL && strcmp(backend, ggml_backend_reg_get_name(i)) != 0) {
2261+
if (backend_filter != NULL && strcmp(backend_filter, ggml_backend_reg_get_name(i)) != 0) {
22622262
printf(" Skipping\n");
22632263
n_ok++;
22642264
continue;
22652265
}
22662266

22672267
ggml_backend_t backend = ggml_backend_reg_init_backend(i, NULL);
22682268
GGML_ASSERT(backend != NULL);
2269+
2270+
if (backend_filter == NULL && ggml_backend_is_cpu(backend)) {
2271+
printf(" Skipping CPU backend\n");
2272+
ggml_backend_free(backend);
2273+
n_ok++;
2274+
continue;
2275+
}
2276+
22692277
printf(" Backend name: %s\n", ggml_backend_name(backend));
22702278

2271-
bool ok = test_backend(backend, mode, op_name);
2279+
bool ok = test_backend(backend, mode, op_name_filter);
22722280

22732281
printf(" Backend %s: ", ggml_backend_name(backend));
22742282
if (ok) {

0 commit comments

Comments
 (0)