mirror of https://github.com/google/gemma.cpp.git
Tiny fix: align template parameter order with parameter order.
PiperOrigin-RevId: 718411494
This commit is contained in:
parent
9646edc908
commit
a133b3d062
|
|
@ -190,7 +190,7 @@ HWY_NOINLINE HWY_MAYBE_UNUSED void RMSNorm(const VecT* HWY_RESTRICT x,
|
||||||
}
|
}
|
||||||
|
|
||||||
// Same as RMSNorm, but its HWY_RESTRICT forbids passing the same pointer.
|
// Same as RMSNorm, but its HWY_RESTRICT forbids passing the same pointer.
|
||||||
template <typename VecT, typename WeightT>
|
template <typename WeightT, typename VecT>
|
||||||
HWY_NOINLINE HWY_MAYBE_UNUSED void RMSNormInplace(
|
HWY_NOINLINE HWY_MAYBE_UNUSED void RMSNormInplace(
|
||||||
const WeightT* HWY_RESTRICT weight, VecT* HWY_RESTRICT inout,
|
const WeightT* HWY_RESTRICT weight, VecT* HWY_RESTRICT inout,
|
||||||
const size_t size) {
|
const size_t size) {
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue