[Mlir-commits] [mlir] [MLIR][NVVM] Add explicit aligned attribute to nvvm.barrier and nvvm.barrier.reduction (PR #200745)
Guray Ozen
llvmlistbot at llvm.org
Mon Jun 1 02:47:08 PDT 2026
================
@@ -1184,18 +1187,21 @@ def NVVM_BarrierReductionOp :
per-thread predicates.
- `reductionPredicate`: The per-thread i32 predicate. It is compared against
zero to form the i1 value fed into the reduction.
+ - `aligned`: Selects between the `.aligned` and non-`.aligned` forms of the
+ underlying `@llvm.nvvm.barrier.cta.red.*` intrinsic family. Defaults to
+ true.
The result is the i32 reduction value computed across all threads
- participating in the barrier. This op always lowers to the aligned form of
- the `@llvm.nvvm.barrier.cta.red.*` intrinsic family.
+ participating in the barrier.
[For more information, see PTX ISA](https://docs.nvidia.com/cuda/parallel-thread-execution/index.html#parallel-synchronization-and-communication-instructions-bar)
}];
let arguments = (ins
Optional<I32>:$barrierId,
BarrierReductionAttr:$reductionOp,
- I32:$reductionPredicate);
+ I32:$reductionPredicate,
+ DefaultValuedAttr<BoolAttr, "true">:$aligned);
----------------
grypp wrote:
same goes here
https://github.com/llvm/llvm-project/pull/200745
More information about the Mlir-commits
mailing list