Home
last modified time | relevance | path

Searched refs:gpuFuncOp (Results 1 – 1 of 1) sorted by relevance

/external/llvm-project/mlir/lib/Conversion/GPUCommon/
DGPUOpsLowering.h31 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()
[all …]