Fix analysis insertion of anything

EnzymeAD · Sep 18, 2023 · 50de4a2 · 50de4a2
1 parent fd06730
commit 50de4a2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/enzyme/Enzyme/TypeAnalysis/TypeTree.h b/enzyme/Enzyme/TypeAnalysis/TypeTree.h
@@ -981,8 +981,12 @@ class TypeTree : public std::enable_shared_from_this<TypeTree> {
                 return false;
 
               // Inserting say a [0]:anything into a [-1]:Float
-              if (CT == BaseType::Anything)
+              if (CT == BaseType::Anything) {
+                // If both at same index, remove old index
+                if (newMoreGeneralThanOld)
+                  toremove.insert(pair.first);
                 continue;
+              }
 
               // Otherwise, inserting a non-equivalent pair into a more general
               // slot. This is invalid.