mirror of https://github.com/google/gemma.cpp.git
parent
2e722f14f1
commit
7164a5e844
|
|
@ -943,7 +943,7 @@ HWY_NOINLINE void EmbedToken(int token, size_t batch_idx, size_t pos,
|
||||||
|
|
||||||
template <typename Weights, typename T>
|
template <typename Weights, typename T>
|
||||||
HWY_NOINLINE void ResidualConnection(
|
HWY_NOINLINE void ResidualConnection(
|
||||||
size_t num_interleaved, T* HWY_RESTRICT other, T* HWY_RESTRICT x,
|
size_t num_interleaved, const T* HWY_RESTRICT other, T* HWY_RESTRICT x,
|
||||||
const LayerWeightsPtrs<Weights>* layer_weights, bool is_attention) {
|
const LayerWeightsPtrs<Weights>* layer_weights, bool is_attention) {
|
||||||
// ResidualType::Add
|
// ResidualType::Add
|
||||||
AddFromBatched(num_interleaved, other, x,
|
AddFromBatched(num_interleaved, other, x,
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue