BFDiv byref fix (#1959)

EnzymeAD · Jul 2, 2024 · cb42e3b · cb42e3b
1 parent b7c2553
commit cb42e3b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/enzyme/tools/enzyme-tblgen/blas-tblgen.cpp b/enzyme/tools/enzyme-tblgen/blas-tblgen.cpp
@@ -1820,6 +1820,8 @@ void emit_dag(bool forward, Twine resultVarName, DagInit *ruleDag,
     os << ") subdenomar.push_back(item);\n";
     os << "           assert(subdenomar.size() == 1);\n";
     os << "           subdenom = subdenomar[0];\n";
+    os << "           subdenom = load_if_ref(Builder2, fpType, subdenom, "
+          "byRefFloat);\n";
     os << "           assert(subnum);\n";
     os << "           assert(subdenom);\n";
     os << resultVarName << " = Builder2.CreateFDiv(subnum, subdenom);\n";