[mlir] Improve async parallel for tests + fix typos
authorEugene Zhulenev <ezhulenev@google.com>
Mon, 6 Dec 2021 19:23:53 +0000 (11:23 -0800)
committerEugene Zhulenev <ezhulenev@google.com>
Mon, 6 Dec 2021 21:27:54 +0000 (13:27 -0800)
commit68a7c001ad9ac20bdacbc4e406fa2b4f27a74b13
treeba38f0259fd22f0156e712b04b2316343263ebe1
parent097a1cb1d5ebb3a0ec4bcaed8ba3ff6a8e33c00a
[mlir] Improve async parallel for tests + fix typos

Do load and store to verify that we process each element of the iteration space once.

Reviewed By: cota

Differential Revision: https://reviews.llvm.org/D115152
mlir/lib/Dialect/Async/Transforms/AsyncParallelFor.cpp
mlir/test/Integration/Dialect/Async/CPU/test-async-parallel-for-1d.mlir
mlir/test/Integration/Dialect/Async/CPU/test-async-parallel-for-2d.mlir