1 //===-- NVPTXMCExpr.cpp - NVPTX specific MC expression classes ------------===// 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 #include "NVPTXMCExpr.h" 10 #include "llvm/ADT/StringExtras.h" 11 #include "llvm/MC/MCAssembler.h" 12 #include "llvm/MC/MCContext.h" 13 #include "llvm/Support/Format.h" 14 using namespace llvm; 15 16 #define DEBUG_TYPE "nvptx-mcexpr" 17 18 const NVPTXFloatMCExpr * 19 NVPTXFloatMCExpr::create(VariantKind Kind, const APFloat &Flt, MCContext &Ctx) { 20 return new (Ctx) NVPTXFloatMCExpr(Kind, Flt); 21 } 22 23 void NVPTXFloatMCExpr::printImpl(raw_ostream &OS, const MCAsmInfo *MAI) const { 24 bool Ignored; 25 unsigned NumHex; 26 APFloat APF = getAPFloat(); 27 28 switch (Kind) { 29 default: llvm_unreachable("Invalid kind!"); 30 case VK_NVPTX_HALF_PREC_FLOAT: 31 // ptxas does not have a way to specify half-precision floats. 32 // Instead we have to print and load fp16 constants as .b16 33 OS << "0x"; 34 NumHex = 4; 35 APF.convert(APFloat::IEEEhalf(), APFloat::rmNearestTiesToEven, &Ignored); 36 break; 37 case VK_NVPTX_BFLOAT_PREC_FLOAT: 38 OS << "0x"; 39 NumHex = 4; 40 APF.convert(APFloat::BFloat(), APFloat::rmNearestTiesToEven, &Ignored); 41 break; 42 case VK_NVPTX_SINGLE_PREC_FLOAT: 43 OS << "0f"; 44 NumHex = 8; 45 APF.convert(APFloat::IEEEsingle(), APFloat::rmNearestTiesToEven, &Ignored); 46 break; 47 case VK_NVPTX_DOUBLE_PREC_FLOAT: 48 OS << "0d"; 49 NumHex = 16; 50 APF.convert(APFloat::IEEEdouble(), APFloat::rmNearestTiesToEven, &Ignored); 51 break; 52 } 53 54 APInt API = APF.bitcastToAPInt(); 55 OS << format_hex_no_prefix(API.getZExtValue(), NumHex, /*Upper=*/true); 56 } 57 58 const NVPTXGenericMCSymbolRefExpr* 59 NVPTXGenericMCSymbolRefExpr::create(const MCSymbolRefExpr *SymExpr, 60 MCContext &Ctx) { 61 return new (Ctx) NVPTXGenericMCSymbolRefExpr(SymExpr); 62 } 63 64 void NVPTXGenericMCSymbolRefExpr::printImpl(raw_ostream &OS, 65 const MCAsmInfo *MAI) const { 66 OS << "generic("; 67 SymExpr->print(OS, MAI); 68 OS << ")"; 69 } 70