Searched defs:gpuFunc (Results 1 – 3 of 3) sorted by relevance
96 if (auto gpuFunc = op->template getParentOfType<gpu::GPUFuncOp>()) { in matchAndRewrite() local
83 auto gpuFunc = in genGPUFunc() local 91 static Value genLaunchGPUFunc(OpBuilder &builder, gpu::GPUFuncOp gpuFunc, in genLaunchGPUFunc() argument 273 static void genGPUCode(PatternRewriter &rewriter, gpu::GPUFuncOp gpuFunc, in genGPUCode() argument 1220 auto gpuFunc = genGPUFunc(rewriter, gpuModule, args); matchAndRewrite() local [all...]
106 if (auto gpuFunc = op->template getParentOfType<GPUFuncOp>()) { in getKnownLaunchDim() local