Handle fmuladd (rust-lang#479)

Author: wsmoses

enzyme/Enzyme/AdjointGenerator.h

@@ -2920,6 +2920,7 @@ class AdjointGenerator
       case Intrinsic::round:
       case Intrinsic::sqrt:
       case Intrinsic::nvvm_sqrt_rn_d:
+      case Intrinsic::fmuladd:
       case Intrinsic::fma:
         return;
       default:
@@ -3129,6 +3130,7 @@ class AdjointGenerator
         return;
       }
 
+      case Intrinsic::fmuladd:
       case Intrinsic::fma: {
         if (vdiff && !gutils->isConstantValue(orig_ops[0])) {
           Value *dif0 = Builder2.CreateFMul(
@@ -3378,7 +3380,8 @@ class AdjointGenerator
         if (Intrinsic::isOverloaded(ID))
 #if LLVM_VERSION_MAJOR >= 13
           llvm::errs() << "cannot handle (reverse) unknown intrinsic\n"
-                       << Intrinsic::getName(ID, ArrayRef<Type *>(), nullptr,
+                       << Intrinsic::getName(ID, ArrayRef<Type *>(),
+                                             gutils->oldFunc->getParent(),
                                              nullptr)
                        << "\n"
                        << I;
@@ -3562,6 +3565,7 @@ class AdjointGenerator
         return;
       }
 
+      case Intrinsic::fmuladd:
       case Intrinsic::fma: {
         if (gutils->isConstantInstruction(&I))
           return;
@@ -3838,7 +3842,8 @@ class AdjointGenerator
         if (Intrinsic::isOverloaded(ID))
 #if LLVM_VERSION_MAJOR >= 13
           llvm::errs() << "cannot handle (forward) unknown intrinsic\n"
-                       << Intrinsic::getName(ID, ArrayRef<Type *>(), nullptr,
+                       << Intrinsic::getName(ID, ArrayRef<Type *>(),
+                                             gutils->oldFunc->getParent(),
                                              nullptr)
                        << "\n"
                        << I;

enzyme/Enzyme/DifferentialUseAnalysis.h

@@ -104,7 +104,8 @@ static inline bool is_use_directly_needed_in_reverse(
         II->getIntrinsicID() == Intrinsic::stackrestore) {
       return false;
     }
-    if (II->getIntrinsicID() == Intrinsic::fma) {
+    if (II->getIntrinsicID() == Intrinsic::fma ||
+        II->getIntrinsicID() == Intrinsic::fmuladd) {
       bool needed = false;
       if (II->getArgOperand(0) == val &&
           !gutils->isConstantValue(II->getArgOperand(1)))

enzyme/Enzyme/TypeAnalysis/TypeAnalysis.cpp

@@ -2981,6 +2981,7 @@ void TypeAnalyzer::visitIntrinsicInst(llvm::IntrinsicInst &I) {
         &I);
     return;
 
+  case Intrinsic::fmuladd:
   case Intrinsic::fma:
     // No direction check as always valid
     updateAnalysis(