[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