Lines Matching defs:gpuFunc

83   auto gpuFunc =
85   gpuFunc->setAttr(gpu::GPUDialect::getKernelFuncAttrName(),
87   return gpuFunc;
91 static Value genLaunchGPUFunc(OpBuilder &builder, gpu::GPUFuncOp gpuFunc,
95   Location loc = gpuFunc->getLoc();
102       .create<gpu::LaunchFuncOp>(loc, gpuFunc, gridSize, blckSize,
273 static void genGPUCode(PatternRewriter &rewriter, gpu::GPUFuncOp gpuFunc,
278   Location loc = gpuFunc->getLoc();
279   Block &block = gpuFunc.getBody().front();
324   rewriter.create<gpu::ReturnOp>(gpuFunc->getLoc());
1220     auto gpuFunc = genGPUFunc(rewriter, gpuModule, args);
1221     genGPUCode(rewriter, gpuFunc, forallOp, constants, scalars, buffers);
1230         genLaunchGPUFunc(rewriter, gpuFunc, args, tokens, numThreads);