Skip to content

Commit 0d422ef

Browse files
committed
address review feedbacks
1 parent c78e836 commit 0d422ef

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

mlir/lib/Dialect/GPU/Transforms/KernelOutlining.cpp

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -365,10 +365,8 @@ class GpuKernelOutliningPass
365365
auto funcWalkResult = func.walk([&](gpu::LaunchOp op) {
366366
SetVector<Value> operands;
367367
std::string kernelFnName;
368-
if (op.hasKernelFuncName()) {
369-
kernelFnName = op->getAttrOfType<mlir::SymbolRefAttr>("kernelFunc")
370-
.getRootReference()
371-
.str();
368+
if (op.getKernelFunc()) {
369+
kernelFnName = op.getKernelFunc()->getRootReference().str();
372370
} else {
373371
kernelFnName =
374372
Twine(op->getParentOfType<SymbolOpInterface>().getName(),
@@ -413,11 +411,9 @@ class GpuKernelOutliningPass
413411
auto *context = getOperation().getContext();
414412
OpBuilder builder(context);
415413
std::string kernelModuleName;
416-
if (gpuLaunchOp.hasKernelModuleName()) {
414+
if (gpuLaunchOp.getKernelModule()) {
417415
kernelModuleName =
418-
gpuLaunchOp->getAttrOfType<mlir::SymbolRefAttr>("kernelModule")
419-
.getRootReference()
420-
.str();
416+
gpuLaunchOp.getKernelModule()->getRootReference().str();
421417
} else {
422418
kernelModuleName = kernelFunc.getName();
423419
}

0 commit comments

Comments
 (0)