Skip to content

Commit db4d358

Browse files
committed
Fixed dumb error.
1 parent 3a32ae6 commit db4d358

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

perf/bench_utils.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ function run_benchmark(model, x; cuda=true)
2828
end
2929

3030
ps = Flux.params(model)
31-
y, back = if model isa Flux.Recur && eltype(x) isa AbstractVector
31+
y, back = if model isa Flux.Recur && eltype(x) isa AbstractArray
3232
pullback(() -> sum(sum([model(x_t) for x_t in x])), ps)
3333
else
3434
pullback(() -> sum(model(x)), ps)

0 commit comments

Comments
 (0)