Lines Matching refs:gpuFuncOp
31 auto gpuFuncOp = cast<gpu::GPUFuncOp>(op); in matchAndRewrite() local
32 Location loc = gpuFuncOp.getLoc(); in matchAndRewrite()
35 workgroupBuffers.reserve(gpuFuncOp.getNumWorkgroupAttributions()); in matchAndRewrite()
36 for (auto en : llvm::enumerate(gpuFuncOp.getWorkgroupAttributions())) { in matchAndRewrite()
48 llvm::formatv("__wg_{0}_{1}", gpuFuncOp.getName(), en.index())); in matchAndRewrite()
50 gpuFuncOp.getLoc(), arrayType, /*isConstant=*/false, in matchAndRewrite()
57 auto funcType = typeConverter->convertType(gpuFuncOp.getType()) in matchAndRewrite()
63 gpuFuncOp.front().getNumArguments()); in matchAndRewrite()
65 gpuFuncOp.getType(), /*isVariadic=*/false, signatureConversion); in matchAndRewrite()
70 for (const auto &attr : gpuFuncOp.getAttrs()) { in matchAndRewrite()
78 gpuFuncOp.getLoc(), gpuFuncOp.getName(), funcType, in matchAndRewrite()
90 rewriter.setInsertionPointToStart(&gpuFuncOp.front()); in matchAndRewrite()
91 unsigned numProperArguments = gpuFuncOp.getNumArguments(); in matchAndRewrite()
110 Value attribution = gpuFuncOp.getWorkgroupAttributions()[en.index()]; in matchAndRewrite()
119 gpuFuncOp.getNumWorkgroupAttributions(); in matchAndRewrite()
121 for (auto en : llvm::enumerate(gpuFuncOp.getPrivateAttributions())) { in matchAndRewrite()
134 gpuFuncOp.getLoc(), int64Ty, in matchAndRewrite()
137 gpuFuncOp.getLoc(), ptrType, numElements, /*alignment=*/0); in matchAndRewrite()
146 rewriter.inlineRegionBefore(gpuFuncOp.getBody(), llvmFuncOp.getBody(), in matchAndRewrite()
152 rewriter.eraseOp(gpuFuncOp); in matchAndRewrite()