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>
|
||||
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) {
|
||||
// ResidualType::Add
|
||||
AddFromBatched(num_interleaved, other, x,
|
||||
|
|
|
|||
Loading…
Reference in New Issue