[mlir][GPUDialect] Implement memory attributions for LaunchOp
authorFabian Mora <fmorac@udel.edu>
Wed, 26 Apr 2023 22:52:37 +0000 (17:52 -0500)
committermax <maksim.levental@gmail.com>
Wed, 26 Apr 2023 22:53:18 +0000 (17:53 -0500)
commit54e96f4f975ec05f44d998cd13032b1b55dad8ea
tree4252752aa536c0ec50a75dd96cb25cb67d10b74d
parentf223fcf67f00c58f49a759b5d43e82d277a346d7
[mlir][GPUDialect] Implement memory attributions for LaunchOp

Currently memory attributions are not supported for gpu::LaunchOp, this patch implements memory attributions for gpu::LaunchOp and modifies the KernelOutlining pass to make the attributions available in GPUFuncOp.

Reviewed By: makslevental

Differential Revision: https://reviews.llvm.org/D147809
mlir/include/mlir/Dialect/GPU/IR/GPUOps.td
mlir/lib/Dialect/GPU/IR/GPUDialect.cpp
mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp
mlir/test/Dialect/GPU/outlining.mlir