[all-commits] [llvm/llvm-project] f47a38: Add async dependencies support for gpu.launch op

Uday Bondhugula via All-commits all-commits at lists.llvm.org
Thu Apr 21 03:56:28 PDT 2022

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: f47a38f51724fab217838aa09cb029c7e0392285
  Author: Uday Bondhugula <uday at polymagelabs.com>
  Date:   2022-04-21 (Thu, 21 Apr 2022)

  Changed paths:
    M mlir/include/mlir/Dialect/GPU/GPUOps.td
    M mlir/lib/Dialect/GPU/IR/GPUDialect.cpp
    M mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp
    M mlir/test/Dialect/GPU/invalid.mlir
    M mlir/test/Dialect/GPU/ops.mlir
    M mlir/test/Dialect/GPU/outlining.mlir

  Log Message:
  Add async dependencies support for gpu.launch op

Add async dependencies support for gpu.launch op: this allows specifying
a list of async tokens ("streams") as dependencies for the launch.

Update the GPU kernel outlining pass lowering to propagate async
dependencies from gpu.launch to gpu.launch_func op. Previously, a new
stream was being created and destroyed for a kernel launch. The async
deps support allows the kernel launch to be serialized on an existing

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

More information about the All-commits mailing list