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

Internal changes 

PiperOrigin-RevId: 853138600
This commit is contained in:
Krzysztof Rymski 2026-01-07 01:21:02 -08:00 committed by Copybara-Service
parent 1605925d1e
commit 2ee1fac74c
3 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
BUILD.bazel
View File

@ -608,7 +608,9 @@ cc_library(
], ],
deps = [ deps = [
":activations", ":activations",
":basics",
":configs", ":configs",
":kv_cache",
":mat", ":mat",
":matmul", ":matmul",
":matmul_env", ":matmul_env",

4
gemma/flash_attention.cc
View File

@ -22,10 +22,14 @@
#include <cstdlib> #include <cstdlib>
#include <iostream> #include <iostream>
#include <limits> #include <limits>
#include <type_traits>
#include <vector> #include <vector>
#include "compression/types.h" // GEMMA_DISABLED_TARGETS #include "compression/types.h" // GEMMA_DISABLED_TARGETS
#include "gemma/flash_structs.h" #include "gemma/flash_structs.h"
#include "gemma/kv_cache.h"
#include "gemma/query.h"
#include "util/basics.h"
#include "util/threading_context.h" #include "util/threading_context.h"
#include "util/zones.h" #include "util/zones.h"
#include "hwy/base.h" #include "hwy/base.h"

2
util/test_util.h
View File

@ -115,7 +115,7 @@ template <typename T>
void PrintMatPtr(MatPtrT<T> mat) { void PrintMatPtr(MatPtrT<T> mat) {
for (int i = 0; i < mat.Rows(); ++i) { for (int i = 0; i < mat.Rows(); ++i) {
for (int j = 0; j < mat.Cols(); ++j) { for (int j = 0; j < mat.Cols(); ++j) {
std::cerr << mat.Row(i)[j] << " ,"; std::cerr << hwy::ConvertScalarTo<float>(mat.Row(i)[j]) << " ,";
} }
std::cerr << std::endl; std::cerr << std::endl;
} }