1//===- AMDGPUMCExpr.h - AMDGPU 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#ifndef LLVM_LIB_TARGET_AMDGPU_MCTARGETDESC_AMDGPUMCEXPR_H
10#define LLVM_LIB_TARGET_AMDGPU_MCTARGETDESC_AMDGPUMCEXPR_H
11
12#include "llvm/ADT/ArrayRef.h"
13#include "llvm/MC/MCExpr.h"
14
15namespace llvm {
16
17class Function;
18class GCNSubtarget;
19
20/// AMDGPU target specific MCExpr operations.
21///
22/// Takes in a minimum of 1 argument to be used with an operation. The supported
23/// operations are:
24/// - (bitwise) or
25/// - max
26///
27/// \note If the 'or'/'max' operations are provided only a single argument, the
28/// operation will act as a no-op and simply resolve as the provided argument.
29///
30class AMDGPUMCExpr : public MCTargetExpr {
31public:
32 enum VariantKind {
33 AGVK_None,
34 AGVK_Or,
35 AGVK_Max,
36 AGVK_ExtraSGPRs,
37 AGVK_TotalNumVGPRs,
38 AGVK_AlignTo,
39 AGVK_Occupancy
40 };
41
42private:
43 VariantKind Kind;
44 MCContext &Ctx;
45 const MCExpr **RawArgs;
46 ArrayRef<const MCExpr *> Args;
47
48 AMDGPUMCExpr(VariantKind Kind, ArrayRef<const MCExpr *> Args, MCContext &Ctx);
49 ~AMDGPUMCExpr();
50
51 bool evaluateExtraSGPRs(MCValue &Res, const MCAssembler *Asm,
52 const MCFixup *Fixup) const;
53 bool evaluateTotalNumVGPR(MCValue &Res, const MCAssembler *Asm,
54 const MCFixup *Fixup) const;
55 bool evaluateAlignTo(MCValue &Res, const MCAssembler *Asm,
56 const MCFixup *Fixup) const;
57 bool evaluateOccupancy(MCValue &Res, const MCAssembler *Asm,
58 const MCFixup *Fixup) const;
59
60public:
61 static const AMDGPUMCExpr *
62 create(VariantKind Kind, ArrayRef<const MCExpr *> Args, MCContext &Ctx);
63
64 static const AMDGPUMCExpr *createOr(ArrayRef<const MCExpr *> Args,
65 MCContext &Ctx) {
66 return create(Kind: VariantKind::AGVK_Or, Args, Ctx);
67 }
68
69 static const AMDGPUMCExpr *createMax(ArrayRef<const MCExpr *> Args,
70 MCContext &Ctx) {
71 return create(Kind: VariantKind::AGVK_Max, Args, Ctx);
72 }
73
74 static const AMDGPUMCExpr *createExtraSGPRs(const MCExpr *VCCUsed,
75 const MCExpr *FlatScrUsed,
76 bool XNACKUsed, MCContext &Ctx);
77
78 static const AMDGPUMCExpr *createTotalNumVGPR(const MCExpr *NumAGPR,
79 const MCExpr *NumVGPR,
80 MCContext &Ctx);
81
82 static const AMDGPUMCExpr *
83 createAlignTo(const MCExpr *Value, const MCExpr *Align, MCContext &Ctx) {
84 return create(Kind: VariantKind::AGVK_AlignTo, Args: {Value, Align}, Ctx);
85 }
86
87 static const AMDGPUMCExpr *createOccupancy(unsigned InitOcc,
88 const MCExpr *NumSGPRs,
89 const MCExpr *NumVGPRs,
90 const GCNSubtarget &STM,
91 MCContext &Ctx);
92
93 VariantKind getKind() const { return Kind; }
94 const MCExpr *getSubExpr(size_t Index) const;
95
96 void printImpl(raw_ostream &OS, const MCAsmInfo *MAI) const override;
97 bool evaluateAsRelocatableImpl(MCValue &Res, const MCAssembler *Asm,
98 const MCFixup *Fixup) const override;
99 void visitUsedExpr(MCStreamer &Streamer) const override;
100 MCFragment *findAssociatedFragment() const override;
101 void fixELFSymbolsInTLSFixups(MCAssembler &) const override{};
102
103 static bool classof(const MCExpr *E) {
104 return E->getKind() == MCExpr::Target;
105 }
106};
107
108} // end namespace llvm
109
110#endif // LLVM_LIB_TARGET_AMDGPU_MCTARGETDESC_AMDGPUMCEXPR_H
111