Add subtarget check before printing used bytes mask pragma

Author: dakersnar

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

@@ -24,6 +24,7 @@
 #include "llvm/Support/ErrorHandling.h"
 #include "llvm/Support/FormatVariadic.h"
 #include <cctype>
+#include "NVPTXSubtarget.h"
 using namespace llvm;
 
 #define DEBUG_TYPE "asm-printer"
@@ -396,13 +397,14 @@ void NVPTXInstPrinter::printMemOperand(const MCInst *MI, int OpNum,
 }
 
 void NVPTXInstPrinter::printUsedBytesMaskPragma(const MCInst *MI, int OpNum,
-                                                raw_ostream &O) {
+                                                raw_ostream &O,
+                                                const MCSubtargetInfo &STI) {
   auto &Op = MI->getOperand(OpNum);
   assert(Op.isImm() && "Invalid operand");
   uint32_t Imm = (uint32_t)Op.getImm();
-  if (Imm != UINT32_MAX) {
+  if (static_cast<const NVPTXSubtarget &>(STI).hasUsedBytesMaskPragma() &&
+      Imm != UINT32_MAX)
     O << ".pragma \"used_bytes_mask " << format_hex(Imm, 1) << "\";\n\t";
-  }
 }
 
 void NVPTXInstPrinter::printRegisterOrSinkSymbol(const MCInst *MI, int OpNum,

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

@@ -46,7 +46,8 @@ 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);
+  void printUsedBytesMaskPragma(const MCInst *MI, int OpNum, raw_ostream &O,
+                                const MCSubtargetInfo &STI);
   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);

llvm/lib/Target/NVPTX/NVPTXSubtarget.h

@@ -89,6 +89,9 @@ class NVPTXSubtarget : public NVPTXGenSubtargetInfo {
     return SmVersion >= 100 && PTXVersion >= 88 &&
            AS == NVPTXAS::ADDRESS_SPACE_GLOBAL;
   }
+  bool hasUsedBytesMaskPragma() const {
+    return SmVersion >= 50 && PTXVersion >= 83;
+  }
   bool hasAtomAddF64() const { return SmVersion >= 60; }
   bool hasAtomScope() const { return SmVersion >= 60; }
   bool hasAtomBitwise64() const { return SmVersion >= 32; }