[all-commits] [llvm/llvm-project] 49ce40: [mlir] AsyncParallelFor: align block size to be a ...

[Eugene Zhulenev via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 49ce40e9ab255588bf5093f468b30cbb6b99895b
      https://github.com/llvm/llvm-project/commit/49ce40e9ab255588bf5093f468b30cbb6b99895b
  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
    M mlir/test/Integration/Dialect/Async/CPU/test-async-parallel-for-2d.mlir

  Log Message:
  -----------
  [mlir] AsyncParallelFor: align block size to be a multiple of inner loops iterations

Depends On D115263

By aligning block size to inner loop iterations parallel_compute_fn LLVM can later unroll and vectorize some of the inner loops with small number of trip counts. Up to 2x speedup in multiple benchmarks.

Reviewed By: bkramer

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