[OSSACanonicalizeOwned] Only discover defs once.

The utility performs two def-use traversals.  The first determines
liveness from uses.  The second rewrites copies.

Previously, the defs whose uses were analyzed were discovered twice,
once during each traversal.  The non-triviality of the discovery logic
(i.e. the logic determining when to walk into the values produced by the
instructions which were the users of visited uses) opened the
possibility for a divergence between the two discoveries.  This
possibility had indeed been realized--the two traversals didn't visit
exactly the same uses, and issues ensue.

Here, the defs whose uses are analyzed are discovered only once (and not
discarded as their uses are analyzed) during the first traversal.  The
second traversal reuses the defs discovered in the first traversal,
eliminating the possibility of a def discovery difference.

The second traversal is now done in a different order.  This results in
perturbing the SIL in certain cases.

Author: nate-chandler

lib/SILOptimizer/Utils/CanonicalizeOSSALifetime.cpp

@@ -1161,16 +1161,7 @@ void CanonicalizeOSSALifetime::rewriteCopies(
   assert(getCurrentDef()->getOwnershipKind() == OwnershipKind::Owned);
 
   // Shadow discoveredDefs in order to constrain its uses.
-  auto &discoveredDefs = this->discoveredDefs;
-
-  SmallVector<unsigned, 8> indexWorklist;
-  ValueSet visitedDefs(getCurrentDef()->getFunction());
-  auto addDefToWorklist = [&](Def def) {
-    if (!visitedDefs.insert(def.getValue()))
-      return;
-    discoveredDefs.push_back(def);
-    indexWorklist.push_back(discoveredDefs.size() - 1);
-  };
+  const auto &discoveredDefs = this->discoveredDefs;
 
   InstructionSetVector instsToDelete(getCurrentDef()->getFunction());
 
@@ -1180,11 +1171,6 @@ void CanonicalizeOSSALifetime::rewriteCopies(
   // it requires a copy.
   auto visitUse = [&](Operand *use) {
     auto *user = use->getUser();
-    // Recurse through copies.
-    if (auto *copy = dyn_cast<CopyValueInst>(user)) {
-      addDefToWorklist(Def::copy(copy));
-      return true;
-    }
     if (destroys.contains(user)) {
       auto *destroy = cast<DestroyValueInst>(user);
       // If this destroy was marked as a final destroy, ignore it; otherwise,
@@ -1217,13 +1203,8 @@ void CanonicalizeOSSALifetime::rewriteCopies(
     return true;
   };
 
-  discoveredDefs.clear();
-  addDefToWorklist(Def::root(getCurrentDef()));
   // Perform a def-use traversal, visiting each use operand.
-
-  while (!indexWorklist.empty()) {
-    auto index = indexWorklist.pop_back_val();
-    auto def = discoveredDefs[index];
+  for (auto def : discoveredDefs) {
     switch (def) {
     case Def::Kind::BorrowedFrom:
     case Def::Kind::Reborrow:

test/SILOptimizer/copy_propagation_opaque.sil

@@ -242,7 +242,7 @@ bb(%0 : @owned $T):
 // CHECK-ONONE-NEXT:  [[CP:%.*]] = copy_value [[INSTANCE]] : $T
 // CHECK-NEXT:  // function_ref takeMultipleOwned
 // CHECK-NEXT:  function_ref @takeMultipleOwned : $@convention(thin) <τ_0_0> (@in τ_0_0, @in τ_0_0) -> ()
-// CHECK-NEXT:  apply %{{.*}}<T>([[INSTANCE]], [[CP]]) : $@convention(thin) <τ_0_0> (@in τ_0_0, @in τ_0_0) -> ()
+// CHECK-NEXT:  apply %{{.*}}<T>([[CP]], [[INSTANCE]]) : $@convention(thin) <τ_0_0> (@in τ_0_0, @in τ_0_0) -> ()
 // CHECK-ONONE-NEXT:  destroy_value
 // CHECK-NEXT:  tuple ()
 // CHECK-NEXT:  return