We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0ac55f7 commit 2cb1f3aCopy full SHA for 2cb1f3a
BUILD.bazel
@@ -590,6 +590,8 @@ cc_test(
590
],
591
)
592
593
+INTERNAL_DEPS = []
594
+
595
cc_library(
596
name = "attention",
597
srcs = [
@@ -684,9 +686,7 @@ cc_library(
684
686
"@highway//:stats",
685
687
"@highway//:thread_pool",
688
"@highway//hwy/contrib/sort:vqsort",
- ] +
- [
689
- ],
+ ] + INTERNAL_DEPS,
690
691
692
cc_test(
gemma/activations.h
@@ -34,6 +34,8 @@
34
35
namespace gcpp {
36
37
+typedef std::vector<float, hwy::AlignedAllocator<float>> AlignedFloatVector;
38
39
// Returns the scale value to use for the query in the attention computation.
40
// Also called by ops_test.
41
static inline float ChooseQueryScale(const ModelConfig& config) {
0 commit comments