Skip to content

Commit b6f2ff9

Browse files
committed
move list_cpu_variants() to be called directly
1 parent 7745263 commit b6f2ff9

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

tests/test-backend-ops.cpp

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -6958,6 +6958,8 @@ static bool test_cpu_variant(const char * variant_name, const char * op_names_fi
69586958
}
69596959

69606960
static void list_cpu_variants() {
6961+
ggml_backend_load_all();
6962+
69616963
std::unordered_map<std::string, std::string> variant_names;
69626964
for (size_t i = 0; i < ggml_backend_reg_count(); i++) {
69636965
ggml_backend_reg_t reg = ggml_backend_reg_get(i);
@@ -7010,7 +7012,6 @@ int main(int argc, char ** argv) {
70107012
const char * backend_filter = nullptr;
70117013
const char * params_filter = nullptr;
70127014
const char * cpu_variant_name = nullptr;
7013-
bool list_variants_flag = false;
70147015

70157016
for (int i = 1; i < argc; i++) {
70167017
if (strcmp(argv[i], "test") == 0) {
@@ -7058,7 +7059,8 @@ int main(int argc, char ** argv) {
70587059
list_all_ops();
70597060
return 0;
70607061
} else if (strcmp(argv[i], "--list") == 0) {
7061-
list_variants_flag = true;
7062+
list_cpu_variants();
7063+
return 0;
70627064
} else if (strcmp(argv[i], "--variant") == 0) {
70637065
if (i + 1 < argc) {
70647066
cpu_variant_name = argv[++i];
@@ -7085,11 +7087,6 @@ int main(int argc, char ** argv) {
70857087
}
70867088

70877089
if (mode == MODE_CPU_VARIANTS) {
7088-
if (list_variants_flag) {
7089-
list_cpu_variants();
7090-
return 0;
7091-
}
7092-
70937090
if (cpu_variant_name == nullptr) {
70947091
printf("Error: cpu-variants mode requires --variant <name> or --list\n");
70957092
usage(argv);

0 commit comments

Comments
 (0)