Fix build

Author: dakersnar

llvm/lib/Target/NVPTX/MCTargetDesc/NVPTXInstPrinter.cpp

@@ -76,6 +76,7 @@ void NVPTXInstPrinter::printRegName(raw_ostream &OS, MCRegister Reg) {
 void NVPTXInstPrinter::printInst(const MCInst *MI, uint64_t Address,
                                  StringRef Annot, const MCSubtargetInfo &STI,
                                  raw_ostream &OS) {
+  this->STI = &STI;
   printInstruction(MI, Address, OS);
 
   // Next always print the annotation.
@@ -397,12 +398,11 @@ void NVPTXInstPrinter::printMemOperand(const MCInst *MI, int OpNum,
 }
 
 void NVPTXInstPrinter::printUsedBytesMaskPragma(const MCInst *MI, int OpNum,
-                                                raw_ostream &O,
-                                                const MCSubtargetInfo &STI) {
+                                                raw_ostream &O) {
   auto &Op = MI->getOperand(OpNum);
   assert(Op.isImm() && "Invalid operand");
   uint32_t Imm = (uint32_t)Op.getImm();
-  if (static_cast<const NVPTXSubtarget &>(STI).hasUsedBytesMaskPragma() &&
+  if (static_cast<const NVPTXSubtarget &>(*STI).hasUsedBytesMaskPragma() &&
       Imm != UINT32_MAX)
     O << ".pragma \"used_bytes_mask " << format_hex(Imm, 1) << "\";\n\t";
 }

llvm/lib/Target/NVPTX/MCTargetDesc/NVPTXInstPrinter.h

@@ -20,6 +20,8 @@ namespace llvm {
 class MCSubtargetInfo;
 
 class NVPTXInstPrinter : public MCInstPrinter {
+  const MCSubtargetInfo *STI = nullptr;
+
 public:
   NVPTXInstPrinter(const MCAsmInfo &MAI, const MCInstrInfo &MII,
                    const MCRegisterInfo &MRI);
@@ -46,8 +48,7 @@ class NVPTXInstPrinter : public MCInstPrinter {
                     StringRef Modifier = {});
   void printMemOperand(const MCInst *MI, int OpNum, raw_ostream &O,
                        StringRef Modifier = {});
-  void printUsedBytesMaskPragma(const MCInst *MI, int OpNum, raw_ostream &O,
-                                const MCSubtargetInfo &STI);
+  void printUsedBytesMaskPragma(const MCInst *MI, int OpNum, raw_ostream &O);
   void printRegisterOrSinkSymbol(const MCInst *MI, int OpNum, raw_ostream &O);
   void printHexu32imm(const MCInst *MI, int OpNum, raw_ostream &O);
   void printProtoIdent(const MCInst *MI, int OpNum, raw_ostream &O);