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