1//===-- NVPTXMCExpr.h - NVPTX specific MC expression classes ----*- C++ -*-===//
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// Modeled after ARMMCExpr
10
11#ifndef LLVM_LIB_TARGET_NVPTX_NVPTXMCEXPR_H
12#define LLVM_LIB_TARGET_NVPTX_NVPTXMCEXPR_H
13
14#include "llvm/ADT/APFloat.h"
15#include "llvm/MC/MCExpr.h"
16#include <utility>
17
18namespace llvm {
19
20class NVPTXFloatMCExpr : public MCTargetExpr {
21public:
22 enum VariantKind {
23 VK_NVPTX_None,
24 VK_NVPTX_BFLOAT_PREC_FLOAT, // FP constant in bfloat-precision
25 VK_NVPTX_HALF_PREC_FLOAT, // FP constant in half-precision
26 VK_NVPTX_SINGLE_PREC_FLOAT, // FP constant in single-precision
27 VK_NVPTX_DOUBLE_PREC_FLOAT // FP constant in double-precision
28 };
29
30private:
31 const VariantKind Kind;
32 const APFloat Flt;
33
34 explicit NVPTXFloatMCExpr(VariantKind Kind, APFloat Flt)
35 : Kind(Kind), Flt(std::move(Flt)) {}
36
37public:
38 /// @name Construction
39 /// @{
40
41 static const NVPTXFloatMCExpr *create(VariantKind Kind, const APFloat &Flt,
42 MCContext &Ctx);
43
44 static const NVPTXFloatMCExpr *createConstantBFPHalf(const APFloat &Flt,
45 MCContext &Ctx) {
46 return create(Kind: VK_NVPTX_BFLOAT_PREC_FLOAT, Flt, Ctx);
47 }
48
49 static const NVPTXFloatMCExpr *createConstantFPHalf(const APFloat &Flt,
50 MCContext &Ctx) {
51 return create(Kind: VK_NVPTX_HALF_PREC_FLOAT, Flt, Ctx);
52 }
53
54 static const NVPTXFloatMCExpr *createConstantFPSingle(const APFloat &Flt,
55 MCContext &Ctx) {
56 return create(Kind: VK_NVPTX_SINGLE_PREC_FLOAT, Flt, Ctx);
57 }
58
59 static const NVPTXFloatMCExpr *createConstantFPDouble(const APFloat &Flt,
60 MCContext &Ctx) {
61 return create(Kind: VK_NVPTX_DOUBLE_PREC_FLOAT, Flt, Ctx);
62 }
63
64 /// @}
65 /// @name Accessors
66 /// @{
67
68 /// getOpcode - Get the kind of this expression.
69 VariantKind getKind() const { return Kind; }
70
71 /// getSubExpr - Get the child of this expression.
72 APFloat getAPFloat() const { return Flt; }
73
74/// @}
75
76 void printImpl(raw_ostream &OS, const MCAsmInfo *MAI) const override;
77 bool evaluateAsRelocatableImpl(MCValue &Res,
78 const MCAsmLayout *Layout,
79 const MCFixup *Fixup) const override {
80 return false;
81 }
82 void visitUsedExpr(MCStreamer &Streamer) const override {};
83 MCFragment *findAssociatedFragment() const override { return nullptr; }
84
85 // There are no TLS NVPTXMCExprs at the moment.
86 void fixELFSymbolsInTLSFixups(MCAssembler &Asm) const override {}
87
88 static bool classof(const MCExpr *E) {
89 return E->getKind() == MCExpr::Target;
90 }
91};
92
93/// A wrapper for MCSymbolRefExpr that tells the assembly printer that the
94/// symbol should be enclosed by generic().
95class NVPTXGenericMCSymbolRefExpr : public MCTargetExpr {
96private:
97 const MCSymbolRefExpr *SymExpr;
98
99 explicit NVPTXGenericMCSymbolRefExpr(const MCSymbolRefExpr *_SymExpr)
100 : SymExpr(_SymExpr) {}
101
102public:
103 /// @name Construction
104 /// @{
105
106 static const NVPTXGenericMCSymbolRefExpr
107 *create(const MCSymbolRefExpr *SymExpr, MCContext &Ctx);
108
109 /// @}
110 /// @name Accessors
111 /// @{
112
113 /// getOpcode - Get the kind of this expression.
114 const MCSymbolRefExpr *getSymbolExpr() const { return SymExpr; }
115
116 /// @}
117
118 void printImpl(raw_ostream &OS, const MCAsmInfo *MAI) const override;
119 bool evaluateAsRelocatableImpl(MCValue &Res,
120 const MCAsmLayout *Layout,
121 const MCFixup *Fixup) const override {
122 return false;
123 }
124 void visitUsedExpr(MCStreamer &Streamer) const override {};
125 MCFragment *findAssociatedFragment() const override { return nullptr; }
126
127 // There are no TLS NVPTXMCExprs at the moment.
128 void fixELFSymbolsInTLSFixups(MCAssembler &Asm) const override {}
129
130 static bool classof(const MCExpr *E) {
131 return E->getKind() == MCExpr::Target;
132 }
133 };
134} // end namespace llvm
135
136#endif
137

source code of llvm/lib/Target/NVPTX/NVPTXMCExpr.h