fix

EnzymeAD · Aug 8, 2024 · 199c10f · 199c10f
1 parent ea02b79
commit 199c10f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/Enzyme.jl b/src/Enzyme.jl
@@ -1417,9 +1417,9 @@ of shape `size(output)` of values of the input type.
         inshape = size(x)
 
         st = @static if VERSION >= v"1.9"
-            Base.stack(cols)
+            Base.stack(rows)
         else
-            reshape(cat(cols..., dims=length(outshape)), (outshape..., inshape...))
+            reshape(cat(rows..., dims=length(outshape)), (outshape..., inshape...))
         end
 
         st2 = if length(outshape) == 1 || VERSION < v"1.9"
@@ -1470,9 +1470,9 @@ end
     if x isa AbstractArray
         inshape = size(x)
         st = @static if VERSION >= v"1.9"
-            Base.stack(cols)
+            Base.stack(rows)
         else
-            reshape(cat(cols..., dims=length(outshape)), (outshape..., inshape...))
+            reshape(cat(rows..., dims=length(outshape)), (outshape..., inshape...))
         end
 
         st2 = if length(outshape) == 1 || VERSION < v"1.9"