1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
//===-- FIRContext.cpp ----------------------------------------------------===//
//
// 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
//
//===----------------------------------------------------------------------===//
//
// Coding style: https://mlir.llvm.org/getting_started/DeveloperGuide/
//
//===----------------------------------------------------------------------===//
#include "flang/Optimizer/Dialect/Support/FIRContext.h"
#include "flang/Optimizer/Dialect/Support/KindMapping.h"
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
#include "mlir/IR/BuiltinAttributes.h"
#include "mlir/IR/BuiltinOps.h"
#include "llvm/TargetParser/Host.h"
void fir::setTargetTriple(mlir::ModuleOp mod, llvm::StringRef triple) {
auto target = fir::determineTargetTriple(triple);
mod->setAttr(mlir::LLVM::LLVMDialect::getTargetTripleAttrName(),
mlir::StringAttr::get(mod.getContext(), target));
}
llvm::Triple fir::getTargetTriple(mlir::ModuleOp mod) {
if (auto target = mod->getAttrOfType<mlir::StringAttr>(
mlir::LLVM::LLVMDialect::getTargetTripleAttrName()))
return llvm::Triple(target.getValue());
return llvm::Triple(llvm::sys::getDefaultTargetTriple());
}
static constexpr const char *kindMapName = "fir.kindmap";
static constexpr const char *defKindName = "fir.defaultkind";
void fir::setKindMapping(mlir::ModuleOp mod, fir::KindMapping &kindMap) {
auto *ctx = mod.getContext();
mod->setAttr(kindMapName, mlir::StringAttr::get(ctx, kindMap.mapToString()));
auto defs = kindMap.defaultsToString();
mod->setAttr(defKindName, mlir::StringAttr::get(ctx, defs));
}
fir::KindMapping fir::getKindMapping(mlir::ModuleOp mod) {
auto *ctx = mod.getContext();
if (auto defs = mod->getAttrOfType<mlir::StringAttr>(defKindName)) {
auto defVals = fir::KindMapping::toDefaultKinds(defs.getValue());
if (auto maps = mod->getAttrOfType<mlir::StringAttr>(kindMapName))
return fir::KindMapping(ctx, maps.getValue(), defVals);
return fir::KindMapping(ctx, defVals);
}
return fir::KindMapping(ctx);
}
fir::KindMapping fir::getKindMapping(mlir::Operation *op) {
auto moduleOp = mlir::dyn_cast<mlir::ModuleOp>(op);
if (moduleOp)
return getKindMapping(moduleOp);
moduleOp = op->getParentOfType<mlir::ModuleOp>();
return getKindMapping(moduleOp);
}
std::string fir::determineTargetTriple(llvm::StringRef triple) {
// Treat "" or "default" as stand-ins for the default machine.
if (triple.empty() || triple == "default")
return llvm::sys::getDefaultTargetTriple();
// Treat "native" as stand-in for the host machine.
if (triple == "native")
return llvm::sys::getProcessTriple();
// TODO: normalize the triple?
return triple.str();
}
|
