summaryrefslogtreecommitdiff
path: root/examples/quantize-stats
diff options
context:
space:
mode:
Diffstat (limited to 'examples/quantize-stats')
-rw-r--r--examples/quantize-stats/quantize-stats.cpp5
1 files changed, 3 insertions, 2 deletions
diff --git a/examples/quantize-stats/quantize-stats.cpp b/examples/quantize-stats/quantize-stats.cpp
index 085fdde3..6e4f7e1e 100644
--- a/examples/quantize-stats/quantize-stats.cpp
+++ b/examples/quantize-stats/quantize-stats.cpp
@@ -282,8 +282,9 @@ int main(int argc, char ** argv) {
break;
}
int j;
- for (j = 0; j < GGML_TYPE_COUNT && strcmp(argv[i], ggml_type_name((ggml_type) j)) != 0; j++) {
- // find match
+ for (j = 0; j < GGML_TYPE_COUNT; ++j) {
+ const auto * name = ggml_type_name((ggml_type) j);
+ if (name && strcmp(argv[i], name) == 0) break;
}
if (j < GGML_TYPE_COUNT) {
params.include_types.push_back((ggml_type) j);