Test enemble_ad fails with JULIA_NUM_THREADS > 1

[lungd]

When I set JULIA_NUM_THREADS=4 and run enemble_ad, I get the following error.

DimensionMismatch("tried to assign 2×101 array to 2×101 destination")
in eval at base/boot.jl:360 
in top-level scope at base/timing.jl:210
in gradient at Zygote/RxTZu/src/compiler/interface.jl:58
in pullback at Zygote/RxTZu/src/compiler/interface.jl:40
in _pullback at Zygote/RxTZu/src/compiler/interface.jl:34
in _pullback at Zygote/RxTZu/src/compiler/interface2.jl
in _pullback at half-cheetah-flux.jl:106 
in _pullback at Zygote/RxTZu/src/compiler/interface2.jl
in _pullback at half-cheetah-flux.jl:97 
in _pullback at Zygote/RxTZu/src/compiler/interface2.jl
in _pullback at DiffEqBase/EeX3b/src/solve.jl:93 
in _pullback at ZygoteRules/OjfTt/src/adjoint.jl:57 
in adjoint at Zygote/RxTZu/src/lib/lib.jl:191 
in _apply at base/boot.jl:804
in _pullback at Zygote/RxTZu/src/compiler/interface2.jl
in _pullback at DiffEqBase/EeX3b/src/solve.jl:96 
in _pullback at ZygoteRules/OjfTt/src/adjoint.jl:57 
in adjoint at Zygote/RxTZu/src/lib/lib.jl:191 
in _apply at base/boot.jl:804 
in _pullback at Zygote/RxTZu/src/compiler/interface2.jl
in _pullback at SciMLBase/yh1iq/src/ensemble/basic_ensemble_solve.jl:103 
in _pullback at Zygote/RxTZu/src/compiler/interface2.jl
in _pullback at SciMLBase/yh1iq/src/ensemble/basic_ensemble_solve.jl:108 
in macro expansion at base/timing.jl:287 
in _pullback at SciMLBase/yh1iq/src/ensemble/basic_ensemble_solve.jl:202 
in _pullback at Zygote/RxTZu/src/compiler/interface2.jl
in _pullback at SciMLBase/yh1iq/src/ensemble/basic_ensemble_solve.jl:223 
in _pullback at Zygote/RxTZu/src/compiler/interface2.jl:9
in macro expansion at Zygote/RxTZu/src/compiler/interface2.jl 
in chain_rrule at Zygote/RxTZu/src/compiler/chainrules.jl:89 
in rrule at ChainRules/du1JV/src/rulesets/Base/array.jl:86 
in reduce at base/abstractarray.jl:1569
in _typed_vcat at base/abstractarray.jl:1561
in setindex! at base/abstractarray.jl:1267 
in _setindex! at base/multidimensional.jl:887 
in _unsafe_setindex! at base/multidimensional.jl:896
in setindex_shape_check at base/indices.jl:253 
in throw_setindex_mismatch at base/indices.jl:193