[mlir][nvvm] Introduce redux op
authorGuray Ozen <guray.ozen@gmail.com>
Thu, 19 Jan 2023 09:09:02 +0000 (10:09 +0100)
committerGuray Ozen <guray.ozen@gmail.com>
Fri, 20 Jan 2023 11:14:24 +0000 (12:14 +0100)
commit36663626ee336905745cb1c259b3b65c9ff656bf
tree76c1b3e79a2414f7c72d41cab924316f827aaba3
parent02435a50f22d3dcfdb486280a3022d6d69ca17c8
[mlir][nvvm] Introduce redux op

Ptx model has `redux.sync` that performs reduction operation on the data from each predicated active thread in the thread group. It only is available sm80+.

This revision adds redux as on op to nvvm dialect.

Reviewed By: nicolasvasilache

Differential Revision: https://reviews.llvm.org/D142088
mlir/include/mlir/Dialect/LLVMIR/NVVMOps.td
mlir/lib/Target/LLVMIR/Dialect/NVVM/NVVMToLLVMIRTranslation.cpp
mlir/test/Dialect/LLVMIR/nvvm.mlir