Lines Matching refs:funcOp
72 matchAndRewrite(gpu::GPUFuncOp funcOp, OpAdaptor adaptor,
183 lowerAsEntryFunction(gpu::GPUFuncOp funcOp, TypeConverter &typeConverter, in lowerAsEntryFunction() argument
187 auto fnType = funcOp.getFunctionType(); in lowerAsEntryFunction()
189 funcOp.emitError("SPIR-V lowering only supports entry functions" in lowerAsEntryFunction()
194 funcOp.emitError( in lowerAsEntryFunction()
205 enumerate(funcOp.getFunctionType().getInputs())) { in lowerAsEntryFunction()
211 funcOp.getLoc(), funcOp.getName(), in lowerAsEntryFunction()
214 for (const auto &namedAttr : funcOp->getAttrs()) { in lowerAsEntryFunction()
221 rewriter.inlineRegionBefore(funcOp.getBody(), newFuncOp.getBody(), in lowerAsEntryFunction()
226 rewriter.eraseOp(funcOp); in lowerAsEntryFunction()
242 getDefaultABIAttrs(MLIRContext *context, gpu::GPUFuncOp funcOp, in getDefaultABIAttrs() argument
244 spirv::TargetEnvAttr targetEnv = spirv::lookupTargetEnvOrDefault(funcOp); in getDefaultABIAttrs()
248 for (auto argIndex : llvm::seq<unsigned>(0, funcOp.getNumArguments())) { in getDefaultABIAttrs()
249 if (funcOp.getArgAttrOfType<spirv::InterfaceVarABIAttr>( in getDefaultABIAttrs()
255 if (funcOp.getArgument(argIndex).getType().isIntOrIndexOrFloat()) in getDefaultABIAttrs()
263 gpu::GPUFuncOp funcOp, OpAdaptor adaptor, in matchAndRewrite() argument
265 if (!gpu::GPUDialect::isKernel(funcOp)) in matchAndRewrite()
269 if (failed(getDefaultABIAttrs(rewriter.getContext(), funcOp, argABI))) { in matchAndRewrite()
271 for (auto argIndex : llvm::seq<unsigned>(0, funcOp.getNumArguments())) { in matchAndRewrite()
273 auto abiAttr = funcOp.getArgAttrOfType<spirv::InterfaceVarABIAttr>( in matchAndRewrite()
276 funcOp.emitRemark( in matchAndRewrite()
286 auto entryPointAttr = spirv::lookupEntryPointABI(funcOp); in matchAndRewrite()
288 funcOp.emitRemark("match failure: missing 'spv.entry_point_abi' attribute"); in matchAndRewrite()
292 funcOp, *getTypeConverter(), rewriter, entryPointAttr, argABI); in matchAndRewrite()