Lines Matching defs:topModule

50 static void markAsGPUContainer(ModuleOp topModule) {
51   topModule->setAttr(gpu::GPUDialect::getContainerModuleAttrName(),
52                      UnitAttr::get(topModule->getContext()));
57 static gpu::GPUModuleOp genGPUModule(OpBuilder &builder, ModuleOp topModule) {
58   for (auto op : topModule.getBodyRegion().getOps<gpu::GPUModuleOp>())
60   markAsGPUContainer(topModule);
61   builder.setInsertionPointToStart(topModule.getBody());
62   return builder.create<gpu::GPUModuleOp>(topModule->getLoc(),
1218     ModuleOp topModule = forallOp->getParentOfType<ModuleOp>();
1219     auto gpuModule = genGPUModule(rewriter, topModule);