@@ -366,13 +366,14 @@ class GpuKernelOutliningPass
366
366
SetVector<Value> operands;
367
367
std::string kernelFnName;
368
368
if (op.hasKernelFuncName ()) {
369
- kernelFnName = op->getAttrOfType <mlir::SymbolRefAttr>(" kernelFunc" ).getRootReference ().str ();
370
- llvm::errs () << " use provided kernel func name = " << kernelFnName << " \n " ;
369
+ kernelFnName = op->getAttrOfType <mlir::SymbolRefAttr>(" kernelFunc" )
370
+ .getRootReference ()
371
+ .str ();
371
372
} else {
372
373
kernelFnName =
373
- Twine (op->getParentOfType <SymbolOpInterface>().getName (), " _kernel" )
374
+ Twine (op->getParentOfType <SymbolOpInterface>().getName (),
375
+ " _kernel" )
374
376
.str ();
375
- llvm::errs () << " use default kernel func name = " << kernelFnName << " \n " ;
376
377
}
377
378
378
379
gpu::GPUFuncOp outlinedFunc =
@@ -402,7 +403,8 @@ class GpuKernelOutliningPass
402
403
403
404
private:
404
405
// / Returns a gpu.module containing kernelFunc and all callees (recursive).
405
- gpu::GPUModuleOp createKernelModule (gpu::LaunchOp gpuLaunchOp, gpu::GPUFuncOp kernelFunc,
406
+ gpu::GPUModuleOp createKernelModule (gpu::LaunchOp gpuLaunchOp,
407
+ gpu::GPUFuncOp kernelFunc,
406
408
const SymbolTable &parentSymbolTable) {
407
409
// TODO: This code cannot use an OpBuilder because it must be inserted into
408
410
// a SymbolTable by the caller. SymbolTable needs to be refactored to
@@ -412,23 +414,23 @@ class GpuKernelOutliningPass
412
414
OpBuilder builder (context);
413
415
std::string kernelModuleName;
414
416
if (gpuLaunchOp.hasKernelModuleName ()) {
415
- kernelModuleName = gpuLaunchOp->getAttrOfType <mlir::SymbolRefAttr>(" kernelModule" ).getRootReference ().str ();
416
- llvm::errs () << " use provided kernel module name = " << kernelModuleName << " \n " ;
417
+ kernelModuleName =
418
+ gpuLaunchOp->getAttrOfType <mlir::SymbolRefAttr>(" kernelModule" )
419
+ .getRootReference ()
420
+ .str ();
417
421
} else {
418
422
kernelModuleName = kernelFunc.getName ();
419
- llvm::errs () << " use default kernel module name = " << kernelModuleName << " \n " ;
420
423
}
421
424
422
425
gpu::GPUModuleOp kernelModule;
423
426
// Check if the module already exists in the symbol table
424
- if (auto existingModule = parentSymbolTable. lookup <gpu::GPUModuleOp>(kernelModuleName)) {
425
- llvm::errs () << " Reusing existing kernel module: " << kernelModuleName << " \n " ;
427
+ if (auto existingModule =
428
+ parentSymbolTable. lookup <gpu::GPUModuleOp>( kernelModuleName)) {
426
429
kernelModule = existingModule;
427
430
} else {
428
431
// If not found, create a new GPU module
429
- llvm::errs () << " Creating new kernel module: " << kernelModuleName << " \n " ;
430
432
kernelModule = builder.create <gpu::GPUModuleOp>(kernelFunc.getLoc (),
431
- kernelModuleName);
433
+ kernelModuleName);
432
434
}
433
435
434
436
// If a valid data layout spec was provided, attach it to the kernel module.
@@ -457,8 +459,6 @@ class GpuKernelOutliningPass
457
459
}
458
460
}
459
461
460
- // llvm::errs() << "kernelModule:\n" << kernelModule << "\n";
461
-
462
462
return kernelModule;
463
463
}
464
464
0 commit comments