Lines Matching defs:moduleOp
62 auto targetEnvSupportsKernelCapability = [](gpu::GPUModuleOp moduleOp) {
63 Operation *gpuModule = moduleOp.getOperation();
69 module.walk([&](gpu::GPUModuleOp moduleOp) {
78 if (targetEnvSupportsKernelCapability(moduleOp)) {
79 builder.setInsertionPointToStart(moduleOp.getBody());
81 builder.setInsertionPoint(moduleOp.getOperation());
83 gpuModules.push_back(builder.clone(*moduleOp.getOperation()));
144 module.walk([&](gpu::GPUModuleOp moduleOp) {
145 if (targetEnvSupportsKernelCapability(moduleOp)) {
146 moduleOp.walk([&](gpu::GPUFuncOp funcOp) {