Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Tiny fix: align template parameter order with parameter order.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 718381988
danielkeysers authored and copybara-github committed Jan 22, 2025

Verified

This commit was signed with the committer’s verified signature.
trim21 Trim21
1 parent 9646edc commit 0543b81
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ops/ops-inl.h
Original file line number Diff line number Diff line change
@@ -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.
template <typename VecT, typename WeightT>
template <typename WeightT, typename VecT>
HWY_NOINLINE HWY_MAYBE_UNUSED void RMSNormInplace(
const WeightT* HWY_RESTRICT weight, VecT* HWY_RESTRICT inout,
const size_t size) {

0 comments on commit 0543b81

Please sign in to comment.