Index: llvm/lib/Target/AArch64/AArch64InstrInfo.h
===================================================================
--- llvm/lib/Target/AArch64/AArch64InstrInfo.h
+++ llvm/lib/Target/AArch64/AArch64InstrInfo.h
@@ -62,10 +62,6 @@
   unsigned isStoreToStackSlot(const MachineInstr &MI,
                               int &FrameIndex) const override;
 
-  /// Returns true if there is an extendable register and that the extending
-  /// value is non-zero.
-  static bool hasExtendedReg(const MachineInstr &MI);
-
   /// Does this instruction set its full destination register to zero?
   static bool isGPRZero(const MachineInstr &MI);
 
Index: llvm/lib/Target/AArch64/AArch64InstrInfo.cpp
===================================================================
--- llvm/lib/Target/AArch64/AArch64InstrInfo.cpp
+++ llvm/lib/Target/AArch64/AArch64InstrInfo.cpp
@@ -1938,33 +1938,6 @@
   return true;
 }
 
-/// Return true if this is this instruction has a non-zero immediate
-bool AArch64InstrInfo::hasExtendedReg(const MachineInstr &MI) {
-  switch (MI.getOpcode()) {
-  default:
-    break;
-  case AArch64::ADDSWrx:
-  case AArch64::ADDSXrx:
-  case AArch64::ADDSXrx64:
-  case AArch64::ADDWrx:
-  case AArch64::ADDXrx:
-  case AArch64::ADDXrx64:
-  case AArch64::SUBSWrx:
-  case AArch64::SUBSXrx:
-  case AArch64::SUBSXrx64:
-  case AArch64::SUBWrx:
-  case AArch64::SUBXrx:
-  case AArch64::SUBXrx64:
-    if (MI.getOperand(3).isImm()) {
-      unsigned val = MI.getOperand(3).getImm();
-      return (val != 0);
-    }
-    break;
-  }
-
-  return false;
-}
-
 // Return true if this instruction simply sets its single destination register
 // to zero. This is equivalent to a register rename of the zero-register.
 bool AArch64InstrInfo::isGPRZero(const MachineInstr &MI) {
Index: llvm/lib/Target/AArch64/AArch64SchedPredicates.td
===================================================================
--- llvm/lib/Target/AArch64/AArch64SchedPredicates.td
+++ llvm/lib/Target/AArch64/AArch64SchedPredicates.td
@@ -28,6 +28,10 @@
 
 // Generic predicates.
 
+// Identify arithmetic instructions with extend.
+def IsArithExtPred        : CheckOpcode<[ADDWrx, ADDXrx, ADDXrx64, ADDSWrx, ADDSXrx, ADDSXrx64,
+                                         SUBWrx, SUBXrx, SUBXrx64, SUBSWrx, SUBSXrx, SUBSXrx64]>;
+
 // Identify arithmetic instructions with shift.
 def IsArithShiftPred      : CheckOpcode<[ADDWrs, ADDXrs, ADDSWrs, ADDSXrs,
                                          SUBWrs, SUBXrs, SUBSWrs, SUBSXrs]>;
@@ -71,19 +75,26 @@
 
 // Target predicates.
 
+// Identify arithmetic instructions with an extended register.
+def RegExtendedBody : CheckAll<[IsArithExtPred,
+                                CheckNot<CheckZeroOperand<3>>]>;
+def RegExtendedPred : MCSchedPredicate<RegExtendedBody>;
+def RegExtendedFn   : TIIPredicate<"hasExtendedReg",
+                                   MCReturnStatement<RegExtendedBody>>;
+
 // Identify arithmetic and logic instructions with a shifted register.
-def RegShiftedBody : CheckAll<[IsArithLogicShiftPred,
-                               CheckNot<CheckZeroOperand<3>>]>;
-def RegShiftedPred : MCSchedPredicate<RegShiftedBody>;
-def RegShiftedFn   : TIIPredicate<"hasShiftedReg",
-                                  MCReturnStatement<RegShiftedBody>>;
+def RegShiftedBody  : CheckAll<[IsArithLogicShiftPred,
+                                CheckNot<CheckZeroOperand<3>>]>;
+def RegShiftedPred  : MCSchedPredicate<RegShiftedBody>;
+def RegShiftedFn    : TIIPredicate<"hasShiftedReg",
+                                   MCReturnStatement<RegShiftedBody>>;
 
 // Identify a load or store using the register offset addressing mode
 // with an extended or scaled register.
-def ScaledIdxBody  : CheckAll<[CheckAny<[IsLoadRegOffsetPred,
-                                         IsStoreRegOffsetPred]>,
-                               CheckAny<[CheckNot<CheckMemExtLSL>,
-                                         CheckMemScaled]>]>;
-def ScaledIdxPred  : MCSchedPredicate<ScaledIdxBody>;
-def ScaledIdxFn    : TIIPredicate<"isScaledAddr",
-                                  MCReturnStatement<ScaledIdxBody>>;
+def ScaledIdxBody   : CheckAll<[CheckAny<[IsLoadRegOffsetPred,
+                                          IsStoreRegOffsetPred]>,
+                                CheckAny<[CheckNot<CheckMemExtLSL>,
+                                          CheckMemScaled]>]>;
+def ScaledIdxPred   : MCSchedPredicate<ScaledIdxBody>;
+def ScaledIdxFn     : TIIPredicate<"isScaledAddr",
+                                   MCReturnStatement<ScaledIdxBody>>;
Index: llvm/lib/Target/AArch64/AArch64Schedule.td
===================================================================
--- llvm/lib/Target/AArch64/AArch64Schedule.td
+++ llvm/lib/Target/AArch64/AArch64Schedule.td
@@ -50,9 +50,6 @@
 def WriteSTIdx : SchedWrite; // Store to a register index (maybe scaled).
 def ReadAdrBase : SchedRead; // Read the base resister of a reg-offset LD/ST.
 
-// Predicate for determining when a extendedable register is extended.
-def RegExtendedPred : SchedPredicate<[{TII->hasExtendedReg(*MI)}]>;
-
 // Serialized two-level address load.
 // EXAMPLE: LOADGot
 def WriteLDAdr : WriteSequence<[WriteAdr, WriteLD]>;