fix output when test cli

kendryte · Nov 6, 2023 · 4411d9e · 4411d9e
1 parent 17dbcc4
commit 4411d9e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/Native/src/test_cli.cpp b/src/Native/src/test_cli.cpp
@@ -78,7 +78,7 @@ result<void> run_core(const std::string &kmodel_path,
                            .count() /
                        1e6);
 
-        if (i == (loop_count - 1)) {
+        if (i == (loop_count - 1) && (entry->parameters_size() < bins.size())) {
             if (ret.is_a<tensor>()) {
                 auto output_bin = bins.back();
                 std::ofstream output_stream(output_bin, std::ios::binary);