diff --git a/src/pystencils/backend/ast/vector.py b/src/pystencils/backend/ast/vector.py index 55db67e7c38a377c3a55ab4d784cfea2a523c78d..d7ae8d6a94ddf747eeb8da9b8e97d350691fb2a9 100644 --- a/src/pystencils/backend/ast/vector.py +++ b/src/pystencils/backend/ast/vector.py @@ -41,10 +41,9 @@ class PsVecBroadcast(PsUnOp, PsVectorOp): class PsVecHorizontal(PsBinOp, PsVectorOp): - """Represents a binary operation between a scalar and a vector operand. - With the binary operation not being vectorized, a horizontal reduction - along the lanes of the vector operand is required to extract a scalar value. - The result type will be equal to the scalar operand. + """Perform a horizontal reduction across a vector onto a scalar base value. + + **Example:** vec_horizontal_add(s, v)` will compute `s + v[0] + v[1] + ... + v[n-1]`. Args: scalar_operand: Scalar operand