[all-commits] [llvm/llvm-project] a6628e: [mlir] Async: add automatic reference counting at ...

[Eugene Zhulenev via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: a6628e596e70bf5c31058dec582c8a7907928e98
      https://github.com/llvm/llvm-project/commit/a6628e596e70bf5c31058dec582c8a7907928e98
  Author: Eugene Zhulenev <ezhulenev at google.com>
  Date:   2021-04-12 (Mon, 12 Apr 2021)

  Changed paths:
    M mlir/include/mlir/Dialect/Async/Passes.h
    M mlir/include/mlir/Dialect/Async/Passes.td
    R mlir/lib/Dialect/Async/Transforms/AsyncRefCounting.cpp
    R mlir/lib/Dialect/Async/Transforms/AsyncRefCountingOptimization.cpp
    A mlir/lib/Dialect/Async/Transforms/AsyncRuntimeRefCounting.cpp
    A mlir/lib/Dialect/Async/Transforms/AsyncRuntimeRefCountingOpt.cpp
    M mlir/lib/Dialect/Async/Transforms/CMakeLists.txt
    R mlir/test/Dialect/Async/async-ref-counting-optimization.mlir
    R mlir/test/Dialect/Async/async-ref-counting.mlir
    A mlir/test/Dialect/Async/async-runtime-ref-counting-opt.mlir
    A mlir/test/Dialect/Async/async-runtime-ref-counting.mlir
    M mlir/test/Integration/Dialect/Async/CPU/microbench-linalg-async-parallel-for.mlir
    M mlir/test/Integration/Dialect/Async/CPU/test-async-parallel-for-1d.mlir
    M mlir/test/Integration/Dialect/Async/CPU/test-async-parallel-for-2d.mlir
    M mlir/test/mlir-cpu-runner/async-group.mlir
    M mlir/test/mlir-cpu-runner/async-value.mlir
    M mlir/test/mlir-cpu-runner/async.mlir

  Log Message:
  -----------
  [mlir] Async: add automatic reference counting at async.runtime operations level

Depends On D95311

Previous automatic-ref-counting pass worked with high level async operations (e.g. async.execute), however async values reference counting is a runtime implementation detail.

New pass mostly relies on the save liveness analysis to place drop_ref operations, and does better verification of CFG with different liveIn sets in block successors.

This is almost NFC change. No new reference counting ideas, just a cleanup of the previous version.

Reviewed By: mehdi_amini

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