happyz
/
gemma.cpp
mirror of https://github.com/google/gemma.cpp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
BUILD.bazel
View File

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

5
gemma/benchmark.cc
View File

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