Lines Matching refs:gpuFuncOp
17 GPUFuncOpLowering::matchAndRewrite(gpu::GPUFuncOp gpuFuncOp, OpAdaptor adaptor, in matchAndRewrite() argument
19 Location loc = gpuFuncOp.getLoc(); in matchAndRewrite()
22 workgroupBuffers.reserve(gpuFuncOp.getNumWorkgroupAttributions()); in matchAndRewrite()
23 for (const auto &en : llvm::enumerate(gpuFuncOp.getWorkgroupAttributions())) { in matchAndRewrite()
35 llvm::formatv("__wg_{0}_{1}", gpuFuncOp.getName(), en.index())); in matchAndRewrite()
37 gpuFuncOp.getLoc(), arrayType, /*isConstant=*/false, in matchAndRewrite()
44 auto funcType = typeConverter->convertType(gpuFuncOp.getFunctionType()) in matchAndRewrite()
50 gpuFuncOp.front().getNumArguments()); in matchAndRewrite()
52 gpuFuncOp.getFunctionType(), /*isVariadic=*/false, signatureConversion); in matchAndRewrite()
57 for (const auto &attr : gpuFuncOp->getAttrs()) { in matchAndRewrite()
67 if (gpuFuncOp.isKernel()) in matchAndRewrite()
70 gpuFuncOp.getLoc(), gpuFuncOp.getName(), funcType, in matchAndRewrite()
83 rewriter.setInsertionPointToStart(&gpuFuncOp.front()); in matchAndRewrite()
84 unsigned numProperArguments = gpuFuncOp.getNumArguments(); in matchAndRewrite()
104 Value attribution = gpuFuncOp.getWorkgroupAttributions()[en.index()]; in matchAndRewrite()
112 unsigned numWorkgroupAttributions = gpuFuncOp.getNumWorkgroupAttributions(); in matchAndRewrite()
114 for (const auto &en : llvm::enumerate(gpuFuncOp.getPrivateAttributions())) { in matchAndRewrite()
127 gpuFuncOp.getLoc(), int64Ty, in matchAndRewrite()
130 gpuFuncOp.getLoc(), ptrType, numElements, /*alignment=*/0); in matchAndRewrite()
139 rewriter.inlineRegionBefore(gpuFuncOp.getBody(), llvmFuncOp.getBody(), in matchAndRewrite()
145 rewriter.eraseOp(gpuFuncOp); in matchAndRewrite()