Searched refs:gpuFuncOp (Results 1 – 2 of 2) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUCommon/ |
| H A D | GPUOpsLowering.cpp | 19 Location loc = gpuFuncOp.getLoc(); in matchAndRewrite()
22 workgroupBuffers.reserve(gpuFuncOp.getNumWorkgroupAttributions()); in matchAndRewrite()
37 gpuFuncOp.getLoc(), arrayType, /*isConstant=*/false, in matchAndRewrite()
50 gpuFuncOp.front().getNumArguments()); 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()
127 gpuFuncOp.getLoc(), int64Ty, in matchAndRewrite()
[all …]
|
| H A D | GPUOpsLowering.h | 25 matchAndRewrite(gpu::GPUFuncOp gpuFuncOp, OpAdaptor adaptor,
|