File tree Expand file tree Collapse file tree 1 file changed +4
-8
lines changed
mlir/lib/Dialect/GPU/Transforms Expand file tree Collapse file tree 1 file changed +4
-8
lines changed Original file line number Diff line number Diff 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 }
You can’t perform that action at this time.
0 commit comments