diff options
| author | Alex Zinenko <zinenko@google.com> | 2019-12-06 14:28:54 -0800 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-12-06 14:29:25 -0800 |
| commit | e96150eb46d8c381f11a7333f0384aad0fc8d1b6 (patch) | |
| tree | fb1a8ad40a7ddec80aadfa14d20759ea3ff879d0 /mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp | |
| parent | 883f5557263e5543c2a8cf7621be885760b47d5d (diff) | |
| download | bcm5719-llvm-e96150eb46d8c381f11a7333f0384aad0fc8d1b6.tar.gz bcm5719-llvm-e96150eb46d8c381f11a7333f0384aad0fc8d1b6.zip | |
Replace custom getBody method with an ODS-generated in gpu::LaunchOp
PiperOrigin-RevId: 284262981
Diffstat (limited to 'mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp')
| -rw-r--r-- | mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp b/mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp index 81d585219a1..fbcd18b9535 100644 --- a/mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp +++ b/mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp @@ -115,7 +115,7 @@ static FuncOp outlineKernelFunc(gpu::LaunchOp launchOp) { std::string kernelFuncName = Twine(launchOp.getParentOfType<FuncOp>().getName(), "_kernel").str(); FuncOp outlinedFunc = FuncOp::create(loc, kernelFuncName, type); - outlinedFunc.getBody().takeBody(launchOp.getBody()); + outlinedFunc.getBody().takeBody(launchOp.body()); Builder builder(launchOp.getContext()); outlinedFunc.setAttr(gpu::GPUDialect::getKernelFuncAttrName(), builder.getUnitAttr()); |
