[all-commits] [llvm/llvm-project] 9f151b: [mlir] AsyncParallelFor: sink constants into the p...

[Eugene Zhulenev via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 9f151b784be0d0b27ad29cb24629815701d60481
      https://github.com/llvm/llvm-project/commit/9f151b784be0d0b27ad29cb24629815701d60481
  Author: Eugene Zhulenev <ezhulenev at google.com>
  Date:   2021-12-09 (Thu, 09 Dec 2021)

  Changed paths:
    M mlir/lib/Dialect/Async/Transforms/AsyncParallelFor.cpp
    M mlir/test/Dialect/Async/async-parallel-for-compute-fn.mlir

  Log Message:
  -----------
  [mlir] AsyncParallelFor: sink constants into the parallel compute function

With complex recursive structure of async dispatch function LLVM can't always propagate constants to the parallel_compute_fn and it often prevents optimizations like loop unrolling and vectorization. We help LLVM by pushing known constants into the parallel_compute_fn explicitly.

Reviewed By: bkramer

Differential Revision: https://reviews.llvm.org/D115263