Refactor code to function

Refactor code to function

Author: scottp101

IGC/Compiler/CISACodeGen/PatternMatchPass.cpp

@@ -4948,31 +4948,7 @@ bool CodeGenPatternMatch::MatchWaveShuffleIndex(llvm::GenIntrinsicInst &I) {
 }
 
 bool CodeGenPatternMatch::MatchWaveInstruction(llvm::GenIntrinsicInst &I) {
-  unsigned int helperLaneIndex = 0;
-  switch (I.getIntrinsicID()) {
-  case GenISAIntrinsic::GenISA_WaveAll:
-  case GenISAIntrinsic::GenISA_WaveClusteredBallot:
-    helperLaneIndex = 2;
-    break;
-  case GenISAIntrinsic::GenISA_WaveBallot:
-  case GenISAIntrinsic::GenISA_WaveInverseBallot:
-    helperLaneIndex = 1;
-    break;
-  case GenISAIntrinsic::GenISA_WaveInterleave:
-  case GenISAIntrinsic::GenISA_WaveClustered:
-  case GenISAIntrinsic::GenISA_WaveClusteredPrefix:
-    helperLaneIndex = 3;
-    break;
-  case GenISAIntrinsic::GenISA_WavePrefix:
-  case GenISAIntrinsic::GenISA_WaveClusteredInterleave:
-    helperLaneIndex = 4;
-    break;
-  default:
-    IGC_ASSERT(false);
-    break;
-  }
-  auto helperLaneMode = cast<ConstantInt>(I.getArgOperand(helperLaneIndex));
-  if (int_cast<int>(helperLaneMode->getSExtValue()) == 1) {
+  if (subgroupIntrinsicHasHelperLanes(I)) {
     m_NeedVMask = true;
   }
   return MatchSingleInstruction(I);

IGC/Compiler/CISACodeGen/helper.cpp

@@ -1336,6 +1336,40 @@ bool isSubGroupShuffleVariant(const llvm::Instruction *I) {
   }
 }
 
+bool subgroupIntrinsicHasHelperLanes(const Instruction &I) {
+  const GenIntrinsicInst *GII = dyn_cast<GenIntrinsicInst>(&I);
+  if (!GII)
+    return false;
+
+  unsigned int helperLaneIndex = 0;
+  switch (GII->getIntrinsicID()) {
+  case GenISAIntrinsic::GenISA_WaveAll:
+  case GenISAIntrinsic::GenISA_WaveClusteredBallot:
+  case GenISAIntrinsic::GenISA_WaveBroadcast:
+  case GenISAIntrinsic::GenISA_WaveShuffleIndex:
+    helperLaneIndex = 2;
+    break;
+  case GenISAIntrinsic::GenISA_WaveBallot:
+  case GenISAIntrinsic::GenISA_WaveInverseBallot:
+    helperLaneIndex = 1;
+    break;
+  case GenISAIntrinsic::GenISA_WaveInterleave:
+  case GenISAIntrinsic::GenISA_WaveClustered:
+  case GenISAIntrinsic::GenISA_WaveClusteredPrefix:
+  case GenISAIntrinsic::GenISA_WaveClusteredBroadcast:
+    helperLaneIndex = 3;
+    break;
+  case GenISAIntrinsic::GenISA_WavePrefix:
+  case GenISAIntrinsic::GenISA_WaveClusteredInterleave:
+    helperLaneIndex = 4;
+    break;
+  default:
+    return false;
+  }
+  auto helperLaneMode = cast<ConstantInt>(GII->getArgOperand(helperLaneIndex));
+  return (int_cast<int>(helperLaneMode->getSExtValue()) == 1);
+}
+
 bool hasSubGroupIntrinsicPVC(llvm::Function &F) {
   for (auto &BB : F) {
     for (auto &I : BB) {

IGC/Compiler/CISACodeGen/helper.h

@@ -174,6 +174,7 @@ bool IsSIMDBlockIntrinsic(const llvm::Instruction *inst);
 bool isSubGroupIntrinsic(const llvm::Instruction *I);
 bool isSubGroupIntrinsicPVC(const llvm::Instruction *I);
 bool isSubGroupShuffleVariant(const llvm::Instruction *I);
+bool subgroupIntrinsicHasHelperLanes(const llvm::Instruction &I);
 bool hasSubGroupIntrinsicPVC(llvm::Function &F);
 
 bool isBarrierIntrinsic(const llvm::Instruction *I);