[X86][APX] Check APX features for keeping using non-rex2 register

Previous PRs for disabling APX for relocation:
llvm#136660
llvm#137794
llvm#138500
llvm#139285

Author: fzou1

llvm/lib/Target/X86/X86RegisterInfo.cpp

@@ -123,13 +123,14 @@ X86RegisterInfo::getLargestLegalSuperClass(const TargetRegisterClass *RC,
   if (RC == &X86::GR8_NOREXRegClass)
     return RC;
 
+  const X86Subtarget &Subtarget = MF.getSubtarget<X86Subtarget>();
+
   // Keep using non-rex2 register class when APX feature (EGPR/NDD/NF) is not
   // enabled for relocation.
-  if (!X86EnableAPXForRelocation && isNonRex2RegClass(RC))
+  if ((Subtarget.hasNDD() || Subtarget.hasNF() || Subtarget.hasEGPR()) &&
+      !X86EnableAPXForRelocation && isNonRex2RegClass(RC))
     return RC;
 
-  const X86Subtarget &Subtarget = MF.getSubtarget<X86Subtarget>();
-
   const TargetRegisterClass *Super = RC;
   auto I = RC->superclasses().begin();
   auto E = RC->superclasses().end();