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, const MCAssembler *Asm,
78 const MCFixup *Fixup) const override {
79 return false;
80 }
81 void visitUsedExpr(MCStreamer &Streamer) const override {};
82 MCFragment *findAssociatedFragment() const override { return nullptr; }
83
84 // There are no TLS NVPTXMCExprs at the moment.
85 void fixELFSymbolsInTLSFixups(MCAssembler &Asm) const override {}
86
87 static bool classof(const MCExpr *E) {
88 return E->getKind() == MCExpr::Target;
89 }
90};
91
92/// A wrapper for MCSymbolRefExpr that tells the assembly printer that the
93/// symbol should be enclosed by generic().
94class NVPTXGenericMCSymbolRefExpr : public MCTargetExpr {
95private:
96 const MCSymbolRefExpr *SymExpr;
97
98 explicit NVPTXGenericMCSymbolRefExpr(const MCSymbolRefExpr *_SymExpr)
99 : SymExpr(_SymExpr) {}
100
101public:
102 /// @name Construction
103 /// @{
104
105 static const NVPTXGenericMCSymbolRefExpr
106 *create(const MCSymbolRefExpr *SymExpr, MCContext &Ctx);
107
108 /// @}
109 /// @name Accessors
110 /// @{
111
112 /// getOpcode - Get the kind of this expression.
113 const MCSymbolRefExpr *getSymbolExpr() const { return SymExpr; }
114
115 /// @}
116
117 void printImpl(raw_ostream &OS, const MCAsmInfo *MAI) const override;
118 bool evaluateAsRelocatableImpl(MCValue &Res, const MCAssembler *Asm,
119 const MCFixup *Fixup) const override {
120 return false;
121 }
122 void visitUsedExpr(MCStreamer &Streamer) const override {};
123 MCFragment *findAssociatedFragment() const override { return nullptr; }
124
125 // There are no TLS NVPTXMCExprs at the moment.
126 void fixELFSymbolsInTLSFixups(MCAssembler &Asm) const override {}
127
128 static bool classof(const MCExpr *E) {
129 return E->getKind() == MCExpr::Target;
130 }
131 };
132} // end namespace llvm
133
134#endif
135