Fix test by printing VPRecipeBase

Author: SamTebbs33

llvm/lib/Transforms/Vectorize/VPlan.cpp

@@ -71,13 +71,9 @@ raw_ostream &llvm::operator<<(raw_ostream &OS, const VPValue &V) {
   V.print(OS, SlotTracker);
   return OS;
 }
-raw_ostream &llvm::operator<<(raw_ostream &OS, const VPDef &D) {
-  /// If this def has a single value, we can cast it to an instruction and use
-  /// its plan for the slot tracker.
-  if (const VPValue *Val = D.getVPSingleValue())
-    return OS << *Val;
-  VPSlotTracker SlotTracker(nullptr);
-  D.print(OS, "", SlotTracker);
+raw_ostream &llvm::operator<<(raw_ostream &OS, const VPRecipeBase &R) {
+  VPSlotTracker SlotTracker(R.getParent()->getPlan());
+  R.print(OS, "", SlotTracker);
   return OS;
 }
 #endif

llvm/lib/Transforms/Vectorize/VPlanValue.h

@@ -195,7 +195,7 @@ typedef DenseMap<Value *, VPValue *> Value2VPValueTy;
 typedef DenseMap<VPValue *, Value *> VPValue2ValueTy;
 
 raw_ostream &operator<<(raw_ostream &OS, const VPValue &V);
-raw_ostream &operator<<(raw_ostream &OS, const VPDef &D);
+raw_ostream &operator<<(raw_ostream &OS, const VPRecipeBase &R);
 
 /// This class augments VPValue with operands which provide the inverse def-use
 /// edges from VPValue's users to their defs.

llvm/unittests/Transforms/Vectorize/VPlanTest.cpp

@@ -812,7 +812,7 @@ Successor(s): ir-bb<scalar.header>
   {
     std::string I4Dump;
     raw_string_ostream OS(I4Dump);
-    OS << *cast<VPDef>(I4);
+    OS << *cast<VPValue>(I4);
     EXPECT_EQ("EMIT vp<%5> = mul vp<%3>, vp<%2>", I4Dump);
   }
 }