From 199c10fa6c7ffe1757f8d15e5a0d4aef3a921b4b Mon Sep 17 00:00:00 2001 From: Billy Moses Date: Thu, 8 Aug 2024 13:07:17 -0400 Subject: [PATCH] fix --- src/Enzyme.jl | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/Enzyme.jl b/src/Enzyme.jl index 6a2a35fda4..e412599357 100644 --- 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"