Home
last modified time | relevance | path

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

/llvm-project/mlir/lib/Conversion/GPUToSPIRV/
H A DGPUToSPIRVPass.cpp63 Operation *gpuModule = moduleOp.getOperation(); in runOnOperation() local
64 auto targetAttr = spirv::lookupTargetEnvOrDefault(gpuModule); in runOnOperation()
88 for (Operation *gpuModule : gpuModules) { in runOnOperation()
90 spirv::lookupTargetEnvOrDefault(gpuModule); in runOnOperation()
96 dyn_cast<gpu::GPUModuleOp>(gpuModule)) in runOnOperation()
100 spirv::convertMemRefTypesAndAttrs(gpuModule, converter); in runOnOperation()
105 gpuModule->walk([&target, this](Operation *childOp) { in runOnOperation()
137 if (failed(applyFullConversion(gpuModule, *target, std::move(patterns)))) in runOnOperation()
/llvm-project/mlir/unittests/Target/LLVM/
H A DSerializeROCDLTarget.cpp87 for (auto gpuModule : (*module).getBody()->getOps<gpu::GPUModuleOp>()) { in TEST_F()
89 serializer.serializeToObject(gpuModule, options); in TEST_F()
123 for (auto gpuModule : (*module).getBody()->getOps<gpu::GPUModuleOp>()) { in TEST_F()
125 serializer.serializeToObject(gpuModule, options); in TEST_F()
149 for (auto gpuModule : (*module).getBody()->getOps<gpu::GPUModuleOp>()) { in TEST_F()
151 serializer.serializeToObject(gpuModule, options); in TEST_F()
173 for (auto gpuModule : (*module).getBody()->getOps<gpu::GPUModuleOp>()) {
175 serializer.serializeToObject(gpuModule, options);
203 for (auto gpuModule : (*module).getBody()->getOps<gpu::GPUModuleOp>()) {
205 serializer.serializeToObject(gpuModule, option
[all...]
H A DSerializeNVVMTarget.cpp88 for (auto gpuModule : (*module).getBody()->getOps<gpu::GPUModuleOp>()) { in TEST_F()
90 serializer.serializeToObject(gpuModule, options); in TEST_F()
124 for (auto gpuModule : (*module).getBody()->getOps<gpu::GPUModuleOp>()) { in TEST_F()
126 serializer.serializeToObject(gpuModule, options); in TEST_F()
154 for (auto gpuModule : (*module).getBody()->getOps<gpu::GPUModuleOp>()) { in TEST_F()
156 serializer.serializeToObject(gpuModule, options); in TEST_F()
204 for (auto gpuModule : (*module).getBody()->getOps<gpu::GPUModuleOp>()) {
206 serializer.serializeToObject(gpuModule, options);
276 for (auto gpuModule : (*module).getBody()->getOps<gpu::GPUModuleOp>()) {
278 serializer.serializeToObject(gpuModule, option
[all...]
/llvm-project/mlir/lib/Dialect/GPU/Transforms/
H A DSPIRVAttachTarget.cpp84 getOperation()->walk([&](gpu::GPUModuleOp gpuModule) { in runOnOperation() argument
86 if (!moduleMatcher.empty() && !matcher.match(gpuModule.getName())) in runOnOperation()
90 if (std::optional<ArrayAttr> attrs = gpuModule.getTargets()) in runOnOperation()
96 gpuModule.setTargetsAttr(builder.getArrayAttr(targets)); in runOnOperation()
/llvm-project/mlir/test/lib/Dialect/SPIRV/
H A DTestEntryPointAbi.cpp62 gpu::GPUModuleOp gpuModule = getOperation(); in runOnOperation() local
65 for (gpu::GPUFuncOp gpuFunc : gpuModule.getOps<gpu::GPUFuncOp>()) { in runOnOperation()
/llvm-project/mlir/lib/Conversion/GPUToVulkan/
H A DConvertGPULaunchFuncToVulkanLaunchFunc.cpp
/llvm-project/mlir/lib/Conversion/SPIRVToLLVM/
H A DConvertLaunchFuncToLLVMCalls.cpp289 for (auto gpuModule : in runOnOperation() local
291 gpuModule.erase(); in runOnOperation()
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseGPUCodegen.cpp67 static gpu::GPUFuncOp genGPUFunc(OpBuilder &builder, gpu::GPUModuleOp gpuModule, in genGPUFunc() argument
76 } while (gpuModule.lookupSymbol(kernelName)); in genGPUFunc()
78 builder.setInsertionPointToStart(gpuModule.getBody()); in genGPUFunc()
82 FunctionType type = FunctionType::get(gpuModule->getContext(), argsTp, {}); in genGPUFunc()
84 builder.create<gpu::GPUFuncOp>(gpuModule->getLoc(), kernelName, type); in genGPUFunc()
1219 auto gpuModule = genGPUModule(rewriter, topModule); in matchAndRewrite() local
1220 auto gpuFunc = genGPUFunc(rewriter, gpuModule, args); in matchAndRewrite()