Make AsyncParallelForRewrite parameterizable with a cost model which drives deciding...
authorbakhtiyar <bakhtiyar@x.team>
Sun, 19 Dec 2021 16:35:37 +0000 (08:35 -0800)
committerEugene Zhulenev <ezhulenev@google.com>
Sun, 19 Dec 2021 16:41:01 +0000 (08:41 -0800)
commitec0e4545caa18e0b55e4f25db913a10a4782fdc3
tree9e1c5f823153d87c9946b876fa1b315a49d05166
parent47bd9ebda404d7d3cf051112b02cd74c459ac3e9
Make AsyncParallelForRewrite parameterizable with a cost model which drives deciding the parallelization granularity.

Reviewed By: ezhulenev, mehdi_amini

Differential Revision: https://reviews.llvm.org/D115423
mlir/include/mlir/Dialect/Async/Passes.h
mlir/include/mlir/Dialect/Async/Transforms.h [new file with mode: 0644]
mlir/lib/Dialect/Async/Transforms/AsyncParallelFor.cpp
mlir/test/Dialect/Async/async-parallel-for-compute-fn.mlir
utils/bazel/llvm-project-overlay/mlir/BUILD.bazel