83451d8d4d
Upstreaming clangIR PR: https://github.com/llvm/clangir/pull/2091 This patch adds support for AMDGPU-specific function attributes for HIP kernels Added setTargetAttributes for AMDGPUTargetCIRGenInfo to set kernel attributes Added generic string attribute handler in amendFunction to translate string-values with "cir." prefix function attributes to LLVM function attributes Follows OGCG AMDGPU implementation from "clang/lib/CodeGen/Targets/AMDGPU.cpp".
130 lines
4.9 KiB
C++
130 lines
4.9 KiB
C++
//===----------------------------------------------------------------------===//
|
|
//
|
|
// 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 file implements lowering of CIR attributes and operations directly to
|
|
// LLVMIR.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
|
|
#include "mlir/IR/DialectRegistry.h"
|
|
#include "mlir/Target/LLVMIR/LLVMTranslationInterface.h"
|
|
#include "mlir/Target/LLVMIR/ModuleTranslation.h"
|
|
#include "clang/CIR/Dialect/IR/CIRDialect.h"
|
|
#include "llvm/ADT/ArrayRef.h"
|
|
#include "llvm/IR/Constant.h"
|
|
#include "llvm/IR/GlobalVariable.h"
|
|
|
|
using namespace llvm;
|
|
|
|
namespace cir {
|
|
namespace direct {
|
|
|
|
/// Implementation of the dialect interface that converts CIR attributes to LLVM
|
|
/// IR metadata.
|
|
class CIRDialectLLVMIRTranslationInterface
|
|
: public mlir::LLVMTranslationDialectInterface {
|
|
public:
|
|
using LLVMTranslationDialectInterface::LLVMTranslationDialectInterface;
|
|
|
|
/// Translates the given operation to LLVM IR using the provided IR builder
|
|
/// and saving the state in `moduleTranslation`.
|
|
mlir::LogicalResult convertOperation(
|
|
mlir::Operation *op, llvm::IRBuilderBase &builder,
|
|
mlir::LLVM::ModuleTranslation &moduleTranslation) const final {
|
|
|
|
if (auto cirOp = llvm::dyn_cast<mlir::LLVM::ZeroOp>(op))
|
|
moduleTranslation.mapValue(cirOp.getResult()) =
|
|
llvm::Constant::getNullValue(
|
|
moduleTranslation.convertType(cirOp.getType()));
|
|
|
|
return mlir::success();
|
|
}
|
|
|
|
/// Any named attribute in the CIR dialect, i.e, with name started with
|
|
/// "cir.", will be handled here.
|
|
virtual mlir::LogicalResult amendOperation(
|
|
mlir::Operation *op, llvm::ArrayRef<llvm::Instruction *> instructions,
|
|
mlir::NamedAttribute attribute,
|
|
mlir::LLVM::ModuleTranslation &moduleTranslation) const override {
|
|
if (auto func = dyn_cast<mlir::LLVM::LLVMFuncOp>(op)) {
|
|
if (mlir::failed(
|
|
amendFunction(func, instructions, attribute, moduleTranslation)))
|
|
return mlir::failure();
|
|
} else if (auto mod = dyn_cast<mlir::ModuleOp>(op)) {
|
|
if (mlir::failed(amendModule(mod, attribute, moduleTranslation)))
|
|
return mlir::failure();
|
|
}
|
|
return mlir::success();
|
|
}
|
|
|
|
private:
|
|
// Translate CIR function attributes to LLVM function attributes.
|
|
mlir::LogicalResult
|
|
amendFunction(mlir::LLVM::LLVMFuncOp func,
|
|
llvm::ArrayRef<llvm::Instruction *> instructions,
|
|
mlir::NamedAttribute attribute,
|
|
mlir::LLVM::ModuleTranslation &moduleTranslation) const {
|
|
llvm::Function *llvmFunc = moduleTranslation.lookupFunction(func.getName());
|
|
llvm::StringRef attrName = attribute.getName().strref();
|
|
|
|
// Strip the "cir." prefix to get the LLVM attribute name.
|
|
llvm::StringRef llvmAttrName = attrName.substr(strlen("cir."));
|
|
if (auto strAttr = mlir::dyn_cast<mlir::StringAttr>(attribute.getValue()))
|
|
llvmFunc->addFnAttr(llvmAttrName, strAttr.getValue());
|
|
return mlir::success();
|
|
}
|
|
|
|
// Translate CIR's module attributes to LLVM's module metadata
|
|
mlir::LogicalResult
|
|
amendModule(mlir::ModuleOp mod, mlir::NamedAttribute attribute,
|
|
mlir::LLVM::ModuleTranslation &moduleTranslation) const {
|
|
llvm::Module *llvmModule = moduleTranslation.getLLVMModule();
|
|
llvm::LLVMContext &llvmContext = llvmModule->getContext();
|
|
|
|
if (attribute.getName() == "cir.amdhsa_code_object_version") {
|
|
if (auto intAttr =
|
|
mlir::dyn_cast<mlir::IntegerAttr>(attribute.getValue())) {
|
|
llvmModule->addModuleFlag(llvm::Module::Error,
|
|
"amdhsa_code_object_version",
|
|
static_cast<uint32_t>(intAttr.getInt()));
|
|
}
|
|
}
|
|
|
|
if (attribute.getName() == "cir.amdgpu_printf_kind") {
|
|
if (auto strAttr =
|
|
mlir::dyn_cast<mlir::StringAttr>(attribute.getValue())) {
|
|
llvm::MDString *mdStr =
|
|
llvm::MDString::get(llvmContext, strAttr.getValue());
|
|
llvmModule->addModuleFlag(llvm::Module::Error, "amdgpu_printf_kind",
|
|
mdStr);
|
|
}
|
|
}
|
|
|
|
return mlir::success();
|
|
}
|
|
};
|
|
|
|
void registerCIRDialectTranslation(mlir::DialectRegistry ®istry) {
|
|
registry.insert<cir::CIRDialect>();
|
|
registry.addExtension(+[](mlir::MLIRContext *ctx, cir::CIRDialect *dialect) {
|
|
dialect->addInterfaces<CIRDialectLLVMIRTranslationInterface>();
|
|
});
|
|
}
|
|
|
|
} // namespace direct
|
|
} // namespace cir
|
|
|
|
namespace mlir {
|
|
void registerCIRDialectTranslation(mlir::MLIRContext &context) {
|
|
mlir::DialectRegistry registry;
|
|
cir::direct::registerCIRDialectTranslation(registry);
|
|
context.appendDialectRegistry(registry);
|
|
}
|
|
} // namespace mlir
|