- Description: Performs root mean square layer normalization (RMSNorm), a normalization method widely used in large language models. Unlike LayerNorm, RMSNorm omits mean subtraction. The AddRmsNorm operator fuses the preceding Add operator with RMSNorm to reduce data movement operations.
- Formula:
Each operator has calls. First, [object Object] is called to obtain the input parameters and compute the required workspace size based on the process. Then, [object Object] is called to perform computation.
[object Object]
[object Object]
Parameters
[object Object]- [object Object]Atlas inference series products[object Object]:
- The data types of
[object Object],[object Object], and[object Object]cannot be BFLOAT16. - The
[object Object]parameter has no effect in the current product scenario.
- The data types of
- [object Object]Atlas inference series products[object Object]:
Returns
aclnnStatus: status code. For details, see .
The first-phase API implements input parameter verification. The following errors may be thrown.
[object Object]
- Boundary value specifications:
- The output is Inf when the input is Inf.
- The output is NaN when the input is NaN.
- Deterministic computation:
- aclnnInplaceAddRmsNorm defaults to a deterministic implementation.
The following example is for reference only. For details, see .
[object Object]