[Mlir-commits] [mlir] [mlir][GPU] Add `RecursiveMemoryEffects` to `gpu.launch` (PR #75315)

Matthias Springer llvmlistbot at llvm.org
Wed Dec 13 02:59:56 PST 2023


https://github.com/matthias-springer created https://github.com/llvm/llvm-project/pull/75315

Infer the side effects of `gpu.launch` from its body.

>From dcd70f891d3e4b54f1c86fbeb2e74af7066aab7f Mon Sep 17 00:00:00 2001
From: Matthias Springer <springerm at google.com>
Date: Wed, 13 Dec 2023 19:58:43 +0900
Subject: [PATCH] [mlir][GPU] Add `RecursiveMemoryEffects` to `gpu.launch`

Infer the side effects of `gpu.launch` from its body.
---
 mlir/include/mlir/Dialect/GPU/IR/GPUOps.td | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/mlir/include/mlir/Dialect/GPU/IR/GPUOps.td b/mlir/include/mlir/Dialect/GPU/IR/GPUOps.td
index 7cad1cd89fd633..d7ab397fb771cb 100644
--- a/mlir/include/mlir/Dialect/GPU/IR/GPUOps.td
+++ b/mlir/include/mlir/Dialect/GPU/IR/GPUOps.td
@@ -672,7 +672,8 @@ def GPU_LaunchFuncOp :GPU_Op<"launch_func", [
 
 def GPU_LaunchOp : GPU_Op<"launch", [
       AutomaticAllocationScope, AttrSizedOperandSegments, GPU_AsyncOpInterface,
-      DeclareOpInterfaceMethods<InferIntRangeInterface>]>,
+      DeclareOpInterfaceMethods<InferIntRangeInterface>,
+      RecursiveMemoryEffects]>,
     Arguments<(ins Variadic<GPU_AsyncToken>:$asyncDependencies,
                Index:$gridSizeX, Index:$gridSizeY, Index:$gridSizeZ,
                Index:$blockSizeX, Index:$blockSizeY, Index:$blockSizeZ,



More information about the Mlir-commits mailing list