[all-commits] [llvm/llvm-project] f57b24: [mlir:Async] Add an async reference counting pass ...

[Eugene Zhulenev via All-commits]

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

  Changed paths:
    M mlir/include/mlir/Dialect/Async/Passes.h
    M mlir/include/mlir/Dialect/Async/Passes.td
    M mlir/lib/Dialect/Async/Transforms/AsyncRuntimeRefCounting.cpp
    A mlir/test/Dialect/Async/async-runtime-policy-based-ref-counting.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

  Log Message:
  -----------
  [mlir:Async] Add an async reference counting pass based on the user defined policy

Depends On D104999

Automatic reference counting based on the liveness analysis can add a lot of reference counting overhead at runtime. If the IR is known to be constrained to few particular "shapes", it's much more efficient to provide a custom reference counting policy that will specify where it is required to update the async value reference count.

Reviewed By: mehdi_amini

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