Fix concat

EnzymeAD · Sep 20, 2023 · e124a85 · e124a85
1 parent 42d47f2
commit e124a85
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/enzyme/tools/enzyme-tblgen/blas-tblgen.cpp b/enzyme/tools/enzyme-tblgen/blas-tblgen.cpp
@@ -918,7 +918,13 @@ void rev_call_arg(DagInit *ruleDag, Rule &rule, size_t actArg, size_t pos,
         return;
       }
       if (Def->getName() == "Concat") {
-        os << "concat_values(";
+        os << "concat_values<";
+        for (size_t i = 0; i < Dag->getNumArgs(); i++) {
+          if (i != 0)
+            os << ", ";
+          os << "ArrayRef<Value*>";
+        }
+        os << ">(";
         for (size_t i = 0; i < Dag->getNumArgs(); i++) {
           if (i != 0)
             os << ", ";