[all-commits] [llvm/llvm-project] b845ad: [mlir][gpu] Add `subgroup_reduce` operation
Ivan Butygin via All-commits
all-commits at lists.llvm.org
Tue Oct 11 02:49:20 PDT 2022
Branch: refs/heads/main
Home: https://github.com/llvm/llvm-project
Commit: b845addae89b6940c1af3c453aab914c6d170d20
https://github.com/llvm/llvm-project/commit/b845addae89b6940c1af3c453aab914c6d170d20
Author: Ivan Butygin <ivan.butygin at gmail.com>
Date: 2022-10-11 (Tue, 11 Oct 2022)
Changed paths:
M mlir/include/mlir/Dialect/GPU/IR/GPUOps.td
M mlir/lib/Dialect/GPU/IR/GPUDialect.cpp
M mlir/test/Dialect/GPU/invalid.mlir
M mlir/test/Dialect/GPU/ops.mlir
Log Message:
-----------
[mlir][gpu] Add `subgroup_reduce` operation
Introduce `subgroup_reduce` operation, similar to `all_reduce`, but operating on subgroup scope instead of workgroup.
It is intended as low-level building block for more high level abstractions (e.g for workgroup-wide `all_reduce` ops).
Only introduce version taking reduce operation enum for simplicity sake.
Differential Revision: https://reviews.llvm.org/D135323
More information about the All-commits
mailing list