move dot product requirements out of switch

Author: inbelic

llvm/lib/Target/SPIRV/SPIRVModuleAnalysis.cpp

@@ -1007,6 +1007,32 @@ void addOpAccessChainReqs(const MachineInstr &Instr,
   }
 }
 
+static void AddDotProductRequirements(const MachineInstr &MI,
+                                      SPIRV::RequirementHandler &Reqs,
+                                      const SPIRVSubtarget &ST) {
+  if (ST.canUseExtension(SPIRV::Extension::SPV_KHR_integer_dot_product))
+    Reqs.addExtension(SPIRV::Extension::SPV_KHR_integer_dot_product);
+  Reqs.addCapability(SPIRV::Capability::DotProductKHR);
+
+  const MachineRegisterInfo &MRI = MI.getMF()->getRegInfo();
+  const MachineInstr *InstrPtr = &MI;
+  assert(MI.getOperand(1).isReg() && "Unexpected operand in dot");
+
+  Register TypeReg = InstrPtr->getOperand(1).getReg();
+  SPIRVType *TypeDef = MRI.getVRegDef(TypeReg);
+  if (TypeDef->getOpcode() == SPIRV::OpTypeInt) {
+    assert(TypeDef->getOperand(1).getImm() == 32);
+    Reqs.addCapability(SPIRV::Capability::DotProductInput4x8BitPackedKHR);
+  } else if (TypeDef->getOpcode() == SPIRV::OpTypeVector) {
+    SPIRVType *ScalarTypeDef = MRI.getVRegDef(TypeDef->getOperand(1).getReg());
+    assert(ScalarTypeDef->getOpcode() == SPIRV::OpTypeInt);
+    auto Capability = ScalarTypeDef->getOperand(1).getImm() == 8
+                          ? SPIRV::Capability::DotProductInput4x8BitKHR
+                          : SPIRV::Capability::DotProductInputAllKHR;
+    Reqs.addCapability(Capability);
+  }
+}
+
 void addInstrRequirements(const MachineInstr &MI,
                           SPIRV::RequirementHandler &Reqs,
                           const SPIRVSubtarget &ST) {
@@ -1384,31 +1410,9 @@ void addInstrRequirements(const MachineInstr &MI,
     }
     break;
   case SPIRV::OpSDot:
-  case SPIRV::OpUDot: {
-    if (ST.canUseExtension(SPIRV::Extension::SPV_KHR_integer_dot_product))
-      Reqs.addExtension(SPIRV::Extension::SPV_KHR_integer_dot_product);
-    Reqs.addCapability(SPIRV::Capability::DotProductKHR);
-
-    const MachineRegisterInfo &MRI = MI.getMF()->getRegInfo();
-    const MachineInstr *InstrPtr = &MI;
-    assert(MI.getOperand(1).isReg() && "Unexpected operand in dot");
-
-    Register TypeReg = InstrPtr->getOperand(1).getReg();
-    SPIRVType *TypeDef = MRI.getVRegDef(TypeReg);
-    if (TypeDef->getOpcode() == SPIRV::OpTypeInt) {
-      assert(TypeDef->getOperand(1).getImm() == 32);
-      Reqs.addCapability(SPIRV::Capability::DotProductInput4x8BitPackedKHR);
-    } else if (TypeDef->getOpcode() == SPIRV::OpTypeVector) {
-      SPIRVType *ScalarTypeDef =
-          MRI.getVRegDef(TypeDef->getOperand(1).getReg());
-      assert(ScalarTypeDef->getOpcode() == SPIRV::OpTypeInt);
-      auto Capability = ScalarTypeDef->getOperand(1).getImm() == 8
-                            ? SPIRV::Capability::DotProductInput4x8BitKHR
-                            : SPIRV::Capability::DotProductInputAllKHR;
-      Reqs.addCapability(Capability);
-    }
+  case SPIRV::OpUDot:
+    AddDotProductRequirements(MI, Reqs, ST);
     break;
-  }
   default:
     break;
   }