b77bac0572
The existing approach to translation to the LLVM IR relies on a single translation supporting the base LLVM dialect, extensible through inheritance to support intrinsic-based dialects also derived from LLVM IR such as NVVM and AVX512. This approach does not scale well as it requires additional translations to be created for each new intrinsic-based dialect and does not allow them to mix in the same module, contrary to the rest of the MLIR infrastructure. Furthermore, OpenMP translation ingrained itself into the main translation mechanism. Start refactoring the translation to LLVM IR to operate using dialect interfaces. Each dialect that contains ops translatable to LLVM IR can implement the interface for translating them, and the top-level translation driver can operate on interfaces without knowing about specific dialects. Furthermore, the delayed dialect registration mechanism allows one to avoid a dependency on LLVM IR in the dialect that is translated to it by implementing the translation as a separate library and only registering it at the client level. This change introduces the new mechanism and factors out the translation of the "main" LLVM dialect. The remaining dialects will follow suit. Reviewed By: nicolasvasilache Differential Revision: https://reviews.llvm.org/D96503
75 lines
3.0 KiB
C++
75 lines
3.0 KiB
C++
//===- mlir-vulkan-runner.cpp - MLIR Vulkan Execution Driver --------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This is a command line utility that executes an MLIR file on the Vulkan by
|
|
// translating MLIR GPU module to SPIR-V and host part to LLVM IR before
|
|
// JIT-compiling and executing the latter.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Conversion/GPUToSPIRV/GPUToSPIRVPass.h"
|
|
#include "mlir/Conversion/GPUToVulkan/ConvertGPUToVulkanPass.h"
|
|
#include "mlir/Conversion/StandardToLLVM/ConvertStandardToLLVMPass.h"
|
|
#include "mlir/Conversion/StandardToSPIRV/StandardToSPIRVPass.h"
|
|
#include "mlir/Dialect/GPU/GPUDialect.h"
|
|
#include "mlir/Dialect/GPU/Passes.h"
|
|
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
|
|
#include "mlir/Dialect/SPIRV/IR/SPIRVDialect.h"
|
|
#include "mlir/Dialect/SPIRV/IR/SPIRVOps.h"
|
|
#include "mlir/Dialect/SPIRV/Transforms/Passes.h"
|
|
#include "mlir/Dialect/StandardOps/IR/Ops.h"
|
|
#include "mlir/ExecutionEngine/JitRunner.h"
|
|
#include "mlir/ExecutionEngine/OptUtils.h"
|
|
#include "mlir/Pass/Pass.h"
|
|
#include "mlir/Pass/PassManager.h"
|
|
#include "mlir/Target/LLVMIR.h"
|
|
#include "llvm/Support/InitLLVM.h"
|
|
#include "llvm/Support/TargetSelect.h"
|
|
|
|
using namespace mlir;
|
|
|
|
static LogicalResult runMLIRPasses(ModuleOp module) {
|
|
PassManager passManager(module.getContext());
|
|
applyPassManagerCLOptions(passManager);
|
|
|
|
passManager.addPass(createGpuKernelOutliningPass());
|
|
passManager.addPass(createLegalizeStdOpsForSPIRVLoweringPass());
|
|
passManager.addPass(createConvertGPUToSPIRVPass());
|
|
OpPassManager &modulePM = passManager.nest<spirv::ModuleOp>();
|
|
modulePM.addPass(spirv::createLowerABIAttributesPass());
|
|
modulePM.addPass(spirv::createUpdateVersionCapabilityExtensionPass());
|
|
passManager.addPass(createConvertGpuLaunchFuncToVulkanLaunchFuncPass());
|
|
LowerToLLVMOptions llvmOptions = {
|
|
/*useBarePtrCallConv =*/false,
|
|
/*emitCWrappers = */ true,
|
|
/*indexBitwidth =*/kDeriveIndexBitwidthFromDataLayout};
|
|
passManager.addPass(createLowerToLLVMPass(llvmOptions));
|
|
passManager.addPass(createConvertVulkanLaunchFuncToVulkanCallsPass());
|
|
return passManager.run(module);
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
llvm::llvm_shutdown_obj x;
|
|
registerPassManagerCLOptions();
|
|
|
|
llvm::InitLLVM y(argc, argv);
|
|
llvm::InitializeNativeTarget();
|
|
llvm::InitializeNativeTargetAsmPrinter();
|
|
mlir::initializeLLVMPasses();
|
|
|
|
mlir::JitRunnerConfig jitRunnerConfig;
|
|
jitRunnerConfig.mlirTransformer = runMLIRPasses;
|
|
|
|
mlir::DialectRegistry registry;
|
|
registry.insert<mlir::LLVM::LLVMDialect, mlir::gpu::GPUDialect,
|
|
mlir::spirv::SPIRVDialect, mlir::StandardOpsDialect>();
|
|
mlir::registerLLVMDialectTranslation(registry);
|
|
|
|
return mlir::JitRunnerMain(argc, argv, registry, jitRunnerConfig);
|
|
}
|