[all-commits] [llvm/llvm-project] d43b23: [mlir:Async] Add the size parameter to the async.g...

[Eugene Zhulenev via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: d43b23608ad664f02f56e965ca78916bde220950
      https://github.com/llvm/llvm-project/commit/d43b23608ad664f02f56e965ca78916bde220950
  Author: Eugene Zhulenev <ezhulenev at google.com>
  Date:   2021-06-25 (Fri, 25 Jun 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Async/IR/AsyncOps.td
    M mlir/include/mlir/ExecutionEngine/AsyncRuntime.h
    M mlir/lib/Conversion/AsyncToLLVM/AsyncToLLVM.cpp
    M mlir/lib/Dialect/Async/Transforms/AsyncParallelFor.cpp
    M mlir/lib/Dialect/Async/Transforms/AsyncToAsyncRuntime.cpp
    M mlir/lib/ExecutionEngine/AsyncRuntime.cpp
    M mlir/test/Conversion/AsyncToLLVM/convert-runtime-to-llvm.mlir
    M mlir/test/Conversion/AsyncToLLVM/convert-to-llvm.mlir
    M mlir/test/Dialect/Async/async-to-async-runtime.mlir
    M mlir/test/Dialect/Async/ops.mlir
    M mlir/test/Dialect/Async/runtime.mlir
    M mlir/test/mlir-cpu-runner/async-error.mlir
    M mlir/test/mlir-cpu-runner/async-group.mlir

  Log Message:
  -----------
  [mlir:Async] Add the size parameter to the async.group

Specify the `!async.group` size (the number of tokens that will be added to it) at construction time. `async.await_all` operation can potentially race with `async.execute` operations that keep updating the group, for this reason it is required to know upfront how many tokens will be added to the group.

Reviewed By: ftynse, herhut

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