[VPlan] Add ExplicitVectorLength to isSingleScalar

michaelmaitland · michaelmaitland · commit b1c58d49d9c5 · 2024-08-30T18:55:21.000-07:00
The docstring of this function: > Returns true if this VPInstruction's operands are single scalars and the > result is also a single scalar. ExplicitVectorLength fits this description. I don't have a test for it now, but it is needed by llvm#106560.
diff --git a/llvm/lib/Transforms/Vectorize/VPlanRecipes.cpp b/llvm/lib/Transforms/Vectorize/VPlanRecipes.cpp
@@ -680,7 +680,8 @@ bool VPInstruction::isVectorToScalar() const {
 }
 
 bool VPInstruction::isSingleScalar() const {
-  return getOpcode() == VPInstruction::ResumePhi;
+  return getOpcode() == VPInstruction::ResumePhi ||
+         getOpcode() == VPInstruction::ExplicitVectorLength;
 }
 
 #if !defined(NDEBUG)

Original file line number	Diff line number	Diff line change
`@@ -680,7 +680,8 @@ bool VPInstruction::isVectorToScalar() const {`
`680`	`680`	`}`
`681`	`681`
`682`	`682`	`bool VPInstruction::isSingleScalar() const {`
`683`		`- return getOpcode() == VPInstruction::ResumePhi;`
	`683`	`+ return getOpcode() == VPInstruction::ResumePhi \|\|`
	`684`	`+ getOpcode() == VPInstruction::ExplicitVectorLength;`
`684`	`685`	`}`
`685`	`686`
`686`	`687`	`#if !defined(NDEBUG)`