[mlir] Async: update condition for dispatching block-aligned compute function
authorEugene Zhulenev <ezhulenev@google.com>
Thu, 17 Feb 2022 18:22:18 +0000 (10:22 -0800)
committerEugene Zhulenev <ezhulenev@google.com>
Wed, 23 Feb 2022 18:29:55 +0000 (10:29 -0800)
commitbeff16f7bd6353054ee7dbf43c6f35082ad61577
treea42fd993b2c01190d8106369425b18dd20ad89c4
parentfc0aa8424ca98da29a9c7aa15b4427d47504ba87
[mlir] Async: update condition for dispatching block-aligned compute function

+ compare block size with the unrollable inner dimension
+ reduce nesting in the code and simplify a bit IR building

Reviewed By: cota

Differential Revision: https://reviews.llvm.org/D120075
mlir/lib/Dialect/Async/Transforms/AsyncParallelFor.cpp
mlir/test/Dialect/Async/async-parallel-for-compute-fn.mlir