quantize : use unused imatrix chunk_size with LLAMA_TRACE

compilade · compilade · commit 2d79a7077cb9 · 2024-09-10T12:09:17.000-04:00
diff --git a/examples/quantize/quantize.cpp b/examples/quantize/quantize.cpp
@@ -220,7 +220,7 @@ static int load_imatrix(const std::string & imatrix_file, std::string & imatrix_
             }
         }
         if (getenv("LLAMA_TRACE")) {
-            printf("%s: loaded data (size = %6d, n_tokens = %6d) for '%s'\n", __func__, int(e.size()), int(max_count), name.c_str());
+            printf("%s: loaded data (size = %6d, n_tokens = %6d, n_chunks = %6d) for '%s'\n", __func__, int(e.size()), int(max_count), int(max_count / chunk_size), name.c_str());
         }
     }
 

Original file line number	Diff line number	Diff line change
`@@ -220,7 +220,7 @@ static int load_imatrix(const std::string & imatrix_file, std::string & imatrix_`
`220`	`220`	`}`
`221`	`221`	`}`
`222`	`222`	`if (getenv("LLAMA_TRACE")) {`
`223`		`- printf("%s: loaded data (size = %6d, n_tokens = %6d) for '%s'\n", __func__, int(e.size()), int(max_count), name.c_str());`
	`223`	`+ printf("%s: loaded data (size = %6d, n_tokens = %6d, n_chunks = %6d) for '%s'\n", __func__, int(e.size()), int(max_count), int(max_count / chunk_size), name.c_str());`
`224`	`224`	`}`
`225`	`225`	`}`
`226`	`226`