encapsulate regmask creation

Author: nocchijiang

llvm/include/llvm/CodeGen/MachineFunction.h

@@ -1148,6 +1148,8 @@ class LLVM_ABI MachineFunction {
   /// Allocate and initialize a register mask with @p NumRegister bits.
   uint32_t *allocateRegMask();
 
+  MutableArrayRef<uint32_t> allocateRegMaskArray();
+
   ArrayRef<int> allocateShuffleMask(ArrayRef<int> Mask);
 
   /// Allocate and construct an extra info structure for a `MachineInstr`.

llvm/lib/CodeGen/MachineFunction.cpp

@@ -623,11 +623,15 @@ const char *MachineFunction::createExternalSymbolName(StringRef Name) {
 }
 
 uint32_t *MachineFunction::allocateRegMask() {
+  return allocateRegMaskArray().data();
+}
+
+MutableArrayRef<uint32_t> MachineFunction::allocateRegMaskArray() {
   unsigned NumRegs = getSubtarget().getRegisterInfo()->getNumRegs();
   unsigned Size = MachineOperand::getRegMaskSize(NumRegs);
   uint32_t *Mask = Allocator.Allocate<uint32_t>(Size);
   memset(Mask, 0, Size * sizeof(Mask[0]));
-  return Mask;
+  return {Mask, Size};
 }
 
 ArrayRef<int> MachineFunction::allocateShuffleMask(ArrayRef<int> Mask) {

llvm/lib/CodeGen/MachineOutliner.cpp

@@ -78,6 +78,7 @@
 #include "llvm/Support/SuffixTree.h"
 #include "llvm/Support/raw_ostream.h"
 #include "llvm/Transforms/Utils/ModuleUtils.h"
+#include <cstdint>
 #include <tuple>
 #include <vector>
 
@@ -1165,15 +1166,13 @@ bool MachineOutliner::outline(
             CallInst->addOperand(
                 MachineOperand::CreateRegMask(*RegMasks.begin()));
           } else {
-            uint32_t *RegMask = MF->allocateRegMask();
-            unsigned NumRegs =
-                MF->getSubtarget().getRegisterInfo()->getNumRegs();
-            unsigned Size = MachineOperand::getRegMaskSize(NumRegs);
-            memset(RegMask, UINT32_MAX, Size * sizeof(RegMask[0]));
+            auto RegMask = MF->allocateRegMaskArray();
+            for (unsigned I = 0; I < RegMask.size(); ++I)
+              RegMask[I] = UINT32_MAX;
             for (const uint32_t *Mask : RegMasks)
-              for (unsigned I = 0; I < Size; ++I)
+              for (unsigned I = 0; I < RegMask.size(); ++I)
                 RegMask[I] &= Mask[I];
-            CallInst->addOperand(MachineOperand::CreateRegMask(RegMask));
+            CallInst->addOperand(MachineOperand::CreateRegMask(RegMask.data()));
           }
         }