Update benchmark with internal init

PiperOrigin-RevId: 641929308
This commit is contained in:
Jan Wassenberg 2024-06-10 09:34:44 -07:00 committed by Copybara-Service
parent 49d814b519
commit c7f5e93136
2 changed files with 6 additions and 0 deletions

View File

@ -214,6 +214,7 @@ cc_binary(
":common",
":cross_entropy",
":gemma_lib",
# "//base",
"//compression:io",
"@hwy//:hwy",
"@hwy//:nanobenchmark",

View File

@ -9,6 +9,7 @@
#include <utility> // std::pair
#include <vector>
// Placeholder for internal header, do not modify.
#include "compression/io.h" // Path
#include "gemma/cross_entropy.h"
#include "gemma/gemma.h"
@ -262,6 +263,10 @@ int BenchmarkTriviaQA(gcpp::Gemma& model, gcpp::InferenceArgs& args,
2b-it-sfp.sbs --goldens_dir "../goldens"
*/
int main(int argc, char** argv) {
{
// Placeholder for internal init, do not modify.
}
gcpp::LoaderArgs loader(argc, argv);
gcpp::InferenceArgs args(argc, argv); // inference
gcpp::AppArgs app(argc, argv);