@@ -2222,8 +2222,8 @@ static void usage(char ** argv) {
2222
2222
2223
2223
int main (int argc, char ** argv) {
2224
2224
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 ;
2227
2227
2228
2228
for (int i = 1 ; i < argc; i++) {
2229
2229
if (strcmp (argv[i], " test" ) == 0 ) {
@@ -2232,14 +2232,14 @@ int main(int argc, char ** argv) {
2232
2232
mode = MODE_PERF;
2233
2233
} else if (strcmp (argv[i], " -o" ) == 0 ) {
2234
2234
if (i + 1 < argc) {
2235
- op_name = argv[++i];
2235
+ op_name_filter = argv[++i];
2236
2236
} else {
2237
2237
usage (argv);
2238
2238
return 1 ;
2239
2239
}
2240
2240
} else if (strcmp (argv[i], " -b" ) == 0 ) {
2241
2241
if (i + 1 < argc) {
2242
- backend = argv[++i];
2242
+ backend_filter = argv[++i];
2243
2243
} else {
2244
2244
usage (argv);
2245
2245
return 1 ;
@@ -2258,17 +2258,25 @@ int main(int argc, char ** argv) {
2258
2258
for (size_t i = 0 ; i < ggml_backend_reg_get_count (); i++) {
2259
2259
printf (" Backend %zu/%zu (%s)\n " , i + 1 , ggml_backend_reg_get_count (), ggml_backend_reg_get_name (i));
2260
2260
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 ) {
2262
2262
printf (" Skipping\n " );
2263
2263
n_ok++;
2264
2264
continue ;
2265
2265
}
2266
2266
2267
2267
ggml_backend_t backend = ggml_backend_reg_init_backend (i, NULL );
2268
2268
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
+
2269
2277
printf (" Backend name: %s\n " , ggml_backend_name (backend));
2270
2278
2271
- bool ok = test_backend (backend, mode, op_name );
2279
+ bool ok = test_backend (backend, mode, op_name_filter );
2272
2280
2273
2281
printf (" Backend %s: " , ggml_backend_name (backend));
2274
2282
if (ok) {
0 commit comments