Tiny fix: align template parameter order with parameter order.

PiperOrigin-RevId: 718411494
This commit is contained in:
Daniel Keysers 2025-01-22 09:12:55 -08:00 committed by Copybara-Service
parent 9646edc908
commit a133b3d062
1 changed files with 1 additions and 1 deletions

View File

@ -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) {