fix constant 1

EnzymeAD · Sep 20, 2023 · 7c42c2f · 7c42c2f
1 parent 5dab863
commit 7c42c2f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/enzyme/Enzyme/BlasDerivatives.td b/enzyme/Enzyme/BlasDerivatives.td
@@ -249,7 +249,7 @@ def gemm : CallBlasPattern<(Op $layout, $transa, $transb, $m, $n, $k, $alpha, $A
                                                   //(Rows $transb, ____, $lda),
                                                   Constant<"1.0">, adj<"B">),
                   /* beta  */ (FrobInnerProd<""> $m, $n, adj<"C">, input<"C">),
-                  /* C     */ (b<"lascl"> $layout, Char<"G">, ConstantInt<0>, ConstantInt<0>, Constant<"1.0">, $beta, $m, $n, adj<"C">, ConstantInt<0>)
+                  /* C     */ (b<"lascl"> $layout, Char<"G">, ConstantInt<0>, ConstantInt<0>, Constant<"1.0">, $beta, $m, $n, adj<"C">)
                   ]
                   >;
 

diff --git a/enzyme/tools/enzyme-tblgen/blas-tblgen.cpp b/enzyme/tools/enzyme-tblgen/blas-tblgen.cpp
@@ -1114,7 +1114,7 @@ void rev_call_args(StringRef argName, Rule &rule, size_t actArg,
   }
   os << "        if (byRef) {\n";
   int n = 0;
-  if (func == "gemv")
+  if (func == "gemv" || func == "lascl")
     n = 1;
   if (func == "gemm")
     n = 2;