other rb_tree mangleing (rust-lang#764)

* other mangleing

* adjust TypeAnalysis to Rb_tree update

* add wip test

* update test and ActivityAnalysis

* fixing older llvm tests

* fix llvm-10 and older

* fix llvm-8 and older

Author: ZuseZ4

enzyme/Enzyme/ActivityAnalysis.cpp

@@ -187,8 +187,12 @@ const std::map<std::string, size_t> MPIInactiveCommAllocators = {
 // Instructions which themselves are inactive
 // the returned value, however, may still be active
 const std::set<std::string> KnownInactiveFunctionInsts = {
-    "__dynamic_cast", "_ZSt18_Rb_tree_decrementPSt18_Rb_tree_node_base",
-    "_ZSt18_Rb_tree_incrementPSt18_Rb_tree_node_base", "jl_ptr_to_array",
+    "__dynamic_cast",
+    "_ZSt18_Rb_tree_decrementPKSt18_Rb_tree_node_base",
+    "_ZSt18_Rb_tree_incrementPKSt18_Rb_tree_node_base",
+    "_ZSt18_Rb_tree_decrementPSt18_Rb_tree_node_base",
+    "_ZSt18_Rb_tree_incrementPSt18_Rb_tree_node_base",
+    "jl_ptr_to_array",
     "jl_ptr_to_array_1d"};
 
 const std::set<std::string> KnownInactiveFunctions = {

enzyme/Enzyme/AdjointGenerator.h

@@ -9012,6 +9012,8 @@ class AdjointGenerator
       // if constant instruction and readonly (thus must be pointer return)
       // and shadow return recomputable from shadow arguments.
       if (funcName == "__dynamic_cast" ||
+          funcName == "_ZSt18_Rb_tree_decrementPKSt18_Rb_tree_node_base" ||
+          funcName == "_ZSt18_Rb_tree_incrementPKSt18_Rb_tree_node_base" ||
           funcName == "_ZSt18_Rb_tree_decrementPSt18_Rb_tree_node_base" ||
           funcName == "_ZSt18_Rb_tree_incrementPSt18_Rb_tree_node_base" ||
           funcName == "jl_ptr_to_array" || funcName == "jl_ptr_to_array_1d") {

enzyme/Enzyme/TypeAnalysis/TypeAnalysis.cpp

@@ -3717,6 +3717,8 @@ void TypeAnalyzer::visitCallInst(CallInst &call) {
       return;
     }
     if (funcName == "__dynamic_cast" ||
+        funcName == "_ZSt18_Rb_tree_decrementPKSt18_Rb_tree_node_base" ||
+        funcName == "_ZSt18_Rb_tree_incrementPKSt18_Rb_tree_node_base" ||
         funcName == "_ZSt18_Rb_tree_decrementPSt18_Rb_tree_node_base" ||
         funcName == "_ZSt18_Rb_tree_incrementPSt18_Rb_tree_node_base") {
       updateAnalysis(&call, TypeTree(BaseType::Pointer).Only(-1), &call);

enzyme/test/Enzyme/ReverseMode/rb_tree.ll

@@ -0,0 +1,38 @@
+; RUN: %opt < %s %loadEnzyme -enzyme -enzyme-preopt=false -mem2reg -instsimplify -adce -correlated-propagation -simplifycfg -S | FileCheck %s
+
+; ModuleID = 'ld-temp.o'
+source_filename = "ld-temp.o"
+target datalayout = "e-m:e-p270:32:32-p271:32:32-p272:64:64-i64:64-f80:128-n8:16:32:64-S128"
+target triple = "x86_64-unknown-linux-gnu"
+
+%"struct.std::_Rb_tree_node_base" = type { i32, %"struct.std::_Rb_tree_node_base"*, %"struct.std::_Rb_tree_node_base"*, %"struct.std::_Rb_tree_node_base"* }
+
+@enzyme_dup = external dso_local local_unnamed_addr global i32, align 4
+@enzyme_const = external dso_local local_unnamed_addr global i32, align 4
+
+declare dso_local i32 @_Z17__enzyme_autodiff(...)
+
+; Function Attrs: mustprogress nofree nounwind readonly willreturn
+declare dso_local %"struct.std::_Rb_tree_node_base"* @_ZSt18_Rb_tree_incrementPKSt18_Rb_tree_node_base(%"struct.std::_Rb_tree_node_base"*) local_unnamed_addr
+
+; Function Attrs: mustprogress nofree nounwind readonly willreturn
+declare dso_local %"struct.std::_Rb_tree_node_base"* @_ZSt18_Rb_tree_decrementPKSt18_Rb_tree_node_base(%"struct.std::_Rb_tree_node_base"*) local_unnamed_addr
+  
+define dso_local i32 @callable(%"struct.std::_Rb_tree_node_base"* %i0) {
+  %i1 = call %"struct.std::_Rb_tree_node_base"* @_ZSt18_Rb_tree_incrementPKSt18_Rb_tree_node_base(%"struct.std::_Rb_tree_node_base"* %i0)
+  %i2 = call %"struct.std::_Rb_tree_node_base"* @_ZSt18_Rb_tree_decrementPKSt18_Rb_tree_node_base(%"struct.std::_Rb_tree_node_base"* %i1)
+  ret i32 0 
+}
+
+define dso_local i32 @main(%"struct.std::_Rb_tree_node_base"* %i0, %"struct.std::_Rb_tree_node_base"* %shaddow) {
+  %id = load i32, i32* @enzyme_dup, align 4
+  %i1 = call i32 (...) @_Z17__enzyme_autodiff(i8* bitcast (i32 (%"struct.std::_Rb_tree_node_base"*)* @callable to i8*), i32 %id, %"struct.std::_Rb_tree_node_base"* nonnull align 8 dereferenceable(24) %i0, %"struct.std::_Rb_tree_node_base"* nonnull align 8 dereferenceable(24) %shaddow)
+  ret i32 0
+}
+
+; CHECK: define internal void @diffecallable(%"struct.std::_Rb_tree_node_base"* %i0, %"struct.std::_Rb_tree_node_base"* %"i0'")
+; CHECK-NEXT: invert:
+; CHECK-NEXT:   %i1 = call %"struct.std::_Rb_tree_node_base"* @_ZSt18_Rb_tree_incrementPKSt18_Rb_tree_node_base(%"struct.std::_Rb_tree_node_base"* %i0)
+; CHECK-NEXT:   %i2 = call %"struct.std::_Rb_tree_node_base"* @_ZSt18_Rb_tree_decrementPKSt18_Rb_tree_node_base(%"struct.std::_Rb_tree_node_base"* %i1)
+; CHECK-NEXT:   ret void
+; CHECK-NEXT: }