[all-commits] [llvm/llvm-project] 0b2137: [mlir] Support pre-existing tokens in 'gpu-async-r...

[Christian Sigg via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 0b21371e1201f23e0f1de2de56ef8eed9c43d6db
      https://github.com/llvm/llvm-project/commit/0b21371e1201f23e0f1de2de56ef8eed9c43d6db
  Author: Christian Sigg <csigg at google.com>
  Date:   2021-06-10 (Thu, 10 Jun 2021)

  Changed paths:
    M mlir/lib/Dialect/GPU/Transforms/AsyncRegionRewriter.cpp
    M mlir/test/Dialect/GPU/async-region.mlir

  Log Message:
  -----------
  [mlir] Support pre-existing tokens in 'gpu-async-region'

Allow gpu ops implementing the async interface to already be async when running the GpuAsyncRegionPass.
That pass threads a 'current token' through a block with ops implementing the gpu async interface.

After this change, existing async ops (returning a !gpu.async.token) set the current token.
Existing synchronous `gpu.wait` ops reset the current token.

Reviewed By: herhut

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