Skip to content

Commit adb57c9

Browse files
committed
move list_cpu_variants() to be called directly
1 parent a509662 commit adb57c9

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
@@ -6932,6 +6932,8 @@ static bool test_cpu_variant(const char * variant_name, const char * op_names_fi
69326932
}
69336933

69346934
static void list_cpu_variants() {
6935+
ggml_backend_load_all();
6936+
69356937
std::unordered_map<std::string, std::string> variant_names;
69366938
for (size_t i = 0; i < ggml_backend_reg_count(); i++) {
69376939
ggml_backend_reg_t reg = ggml_backend_reg_get(i);
@@ -6984,7 +6986,6 @@ int main(int argc, char ** argv) {
69846986
const char * backend_filter = nullptr;
69856987
const char * params_filter = nullptr;
69866988
const char * cpu_variant_name = nullptr;
6987-
bool list_variants_flag = false;
69886989

69896990
for (int i = 1; i < argc; i++) {
69906991
if (strcmp(argv[i], "test") == 0) {
@@ -7032,7 +7033,8 @@ int main(int argc, char ** argv) {
70327033
list_all_ops();
70337034
return 0;
70347035
} else if (strcmp(argv[i], "--list") == 0) {
7035-
list_variants_flag = true;
7036+
list_cpu_variants();
7037+
return 0;
70367038
} else if (strcmp(argv[i], "--variant") == 0) {
70377039
if (i + 1 < argc) {
70387040
cpu_variant_name = argv[++i];
@@ -7059,11 +7061,6 @@ int main(int argc, char ** argv) {
70597061
}
70607062

70617063
if (mode == MODE_CPU_VARIANTS) {
7062-
if (list_variants_flag) {
7063-
list_cpu_variants();
7064-
return 0;
7065-
}
7066-
70677064
if (cpu_variant_name == nullptr) {
70687065
printf("Error: cpu-variants mode requires --variant <name> or --list\n");
70697066
usage(argv);

0 commit comments

Comments
 (0)