Skip to content

Commit 2cb1f3a

Browse files
Krzysztof Rymskicopybara-github
authored andcommitted
Internal changes
PiperOrigin-RevId: 845264767
1 parent 0ac55f7 commit 2cb1f3a

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

BUILD.bazel

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -590,6 +590,8 @@ cc_test(
590590
],
591591
)
592592

593+
INTERNAL_DEPS = []
594+
593595
cc_library(
594596
name = "attention",
595597
srcs = [
@@ -684,9 +686,7 @@ cc_library(
684686
"@highway//:stats",
685687
"@highway//:thread_pool",
686688
"@highway//hwy/contrib/sort:vqsort",
687-
] +
688-
[
689-
],
689+
] + INTERNAL_DEPS,
690690
)
691691

692692
cc_test(

gemma/activations.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,8 @@
3434

3535
namespace gcpp {
3636

37+
typedef std::vector<float, hwy::AlignedAllocator<float>> AlignedFloatVector;
38+
3739
// Returns the scale value to use for the query in the attention computation.
3840
// Also called by ops_test.
3941
static inline float ChooseQueryScale(const ModelConfig& config) {

0 commit comments

Comments
 (0)