Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Conversion/GPUCommon/
H A DGPUOpsLowering.h17 struct GPUFuncOpLowering : ConvertOpToLLVMPattern<gpu::GPUFuncOp> {
20 : ConvertOpToLLVMPattern<gpu::GPUFuncOp>(converter), in GPUFuncOpLowering()
25 matchAndRewrite(gpu::GPUFuncOp gpuFuncOp, OpAdaptor adaptor,
H A DGPUOpsLowering.cpp17 GPUFuncOpLowering::matchAndRewrite(gpu::GPUFuncOp gpuFuncOp, OpAdaptor adaptor, in matchAndRewrite()
60 attr.getName() == gpu::GPUFuncOp::getNumWorkgroupAttributionsAttrName()) in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/GPU/Transforms/
H A DMemoryPromotion.h20 class GPUFuncOp; variable
25 void promoteToWorkgroupMemory(gpu::GPUFuncOp op, unsigned arg);
H A DUtils.h24 class GPUFuncOp; variable
36 gpu::GPUFuncOp outlineKernelFunc(gpu::LaunchOp launchOp, StringRef kernelFnName,
/llvm-project-15.0.7/mlir/test/lib/Dialect/GPU/
H A DTestGpuMemoryPromotion.cpp32 OperationPass<gpu::GPUFuncOp>> {
44 gpu::GPUFuncOp op = getOperation(); in runOnOperation()
/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DKernelOutlining.cpp146 static gpu::GPUFuncOp outlineKernelFuncImpl(gpu::LaunchOp launchOp, in outlineKernelFuncImpl()
167 auto outlinedFunc = builder.create<gpu::GPUFuncOp>(loc, kernelFnName, type); in outlineKernelFuncImpl()
205 gpu::GPUFuncOp mlir::outlineKernelFunc(gpu::LaunchOp launchOp, in outlineKernelFunc()
223 gpu::GPUFuncOp kernelFunc, in convertToLaunchFuncOp()
308 gpu::GPUFuncOp outlinedFunc = in runOnOperation()
335 gpu::GPUModuleOp createKernelModule(gpu::GPUFuncOp kernelFunc, in createKernelModule()
H A DAllReduceLowering.cpp31 GpuAllReduceRewriter(gpu::GPUFuncOp funcOp, gpu::AllReduceOp reduceOp, in GpuAllReduceRewriter()
378 gpu::GPUFuncOp funcOp;
392 : RewritePattern(gpu::GPUFuncOp::getOperationName(), 1, context) {} in GpuAllReduceConversion()
396 auto funcOp = cast<gpu::GPUFuncOp>(op); in matchAndRewrite()
H A DMemoryPromotion.cpp144 void mlir::promoteToWorkgroupMemory(GPUFuncOp op, unsigned arg) { in promoteToWorkgroupMemory()
/llvm-project-15.0.7/mlir/test/lib/Dialect/SPIRV/
H A DTestEntryPointAbi.cpp55 for (gpu::GPUFuncOp gpuFunc : gpuModule.getOps<gpu::GPUFuncOp>()) { in runOnOperation()
/llvm-project-15.0.7/mlir/lib/Dialect/GPU/IR/
H A DGPUDialect.cpp249 auto kernelGPUFunction = dyn_cast<gpu::GPUFuncOp>(kernelFunc); in verifyOperationAttribute()
688 GPUFuncOp kernelFunc, KernelDim3 gridSize, in build()
817 BlockArgument GPUFuncOp::addPrivateAttribution(Type type, Location loc) { in addPrivateAttribution()
823 void GPUFuncOp::build(OpBuilder &builder, OperationState &result, in build()
901 result.addAttribute(GPUFuncOp::getTypeAttrName(), TypeAttr::get(type)); in parse()
907 if (failed(parseAttributions(parser, GPUFuncOp::getWorkgroupKeyword(), in parse()
914 result.addAttribute(GPUFuncOp::getNumWorkgroupAttributionsAttrName(), in parse()
948 void GPUFuncOp::print(OpAsmPrinter &p) { in print()
970 LogicalResult GPUFuncOp::verifyType() { in verifyType()
999 LogicalResult GPUFuncOp::verifyBody() { in verifyBody()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/GPUToSPIRV/
H A DGPUToSPIRV.cpp67 class GPUFuncOpConversion final : public OpConversionPattern<gpu::GPUFuncOp> {
69 using OpConversionPattern<gpu::GPUFuncOp>::OpConversionPattern;
72 matchAndRewrite(gpu::GPUFuncOp funcOp, OpAdaptor adaptor,
183 lowerAsEntryFunction(gpu::GPUFuncOp funcOp, TypeConverter &typeConverter, in lowerAsEntryFunction()
242 getDefaultABIAttrs(MLIRContext *context, gpu::GPUFuncOp funcOp, in getDefaultABIAttrs()
263 gpu::GPUFuncOp funcOp, OpAdaptor adaptor, in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/GPU/IR/
H A DGPUOps.td428 OpBuilder<(ins "GPUFuncOp":$kernelFunc, "KernelDim3":$gridSize,
621 def GPU_ReturnOp : GPU_Op<"return", [HasParent<"GPUFuncOp">, NoSideEffect,