Failed sharding test on macOS

[giordano]

https://github.com/EnzymeAD/Reactant.jl/actions/runs/13359995500/job/37310412074#step:10:449

Sharding Across 8 Devices: Test Failed at /Users/runner/work/Reactant.jl/Reactant.jl/test/sharding.jl:97
  Expression: Array(#= /Users/runner/work/Reactant.jl/Reactant.jl/test/sharding.jl:97 =# @jit(predict(samples_ra, w1_ra, w2_ra))) ≈ predict(samples, w1, w2)
   Evaluated: Float32[-0.9303045 -0.818692 … -0.80645585 -0.80645585; -0.99376154 0.048814937 … 0.02685846 0.02685846; … ; 0.82870674 -0.97798175 … -0.9835431 -0.9835431; 0.9942749 0.7648038 … 0.74527794 0.74527794] ≈ Float32[-0.9303045 -0.818692 … -0.8063114 -0.8063114; -0.9937751 0.048814937 … 0.026370352 0.026370352; … ; 0.82870674 -0.9779308 … -0.98363125 -0.98363125; 0.99430096 0.7648038 … 0.7449523 0.7449523]

Stacktrace:
 [1] macro expansion
   @ ~/hostedtoolcache/julia/1.10.8/x64/share/julia/stdlib/v1.10/Test/src/Test.jl:672 [inlined]
 [2] macro expansion
   @ ~/work/Reactant.jl/Reactant.jl/test/sharding.jl:97 [inlined]
 [3] macro expansion
   @ ~/hostedtoolcache/julia/1.10.8/x64/share/julia/stdlib/v1.10/Test/src/Test.jl:1577 [inlined]
 [4] top-level scope
   @ ~/work/Reactant.jl/Reactant.jl/test/sharding.jl:60

https://github.com/EnzymeAD/Reactant.jl/actions/runs/13359995500/job/37310413722#step:10:538

Sharding Across 8 Devices: Test Failed at /Users/runner/work/Reactant.jl/Reactant.jl/test/sharding.jl:97
  Expression: Array(#= /Users/runner/work/Reactant.jl/Reactant.jl/test/sharding.jl:97 =# @jit(predict(samples_ra, w1_ra, w2_ra))) ≈ predict(samples, w1, w2)
   Evaluated: Float32[-0.9303045 -0.818692 … -0.80645585 -0.80645585; -0.99376154 0.048814937 … 0.02685846 0.02685846; … ; 0.82870674 -0.97798175 … -0.9835431 -0.9835431; 0.9942749 0.7648038 … 0.74527794 0.74527794] ≈ Float32[-0.9303045 -0.818692 … -0.8063114 -0.8063114; -0.9937751 0.048814937 … 0.026370352 0.026370352; … ; 0.82870674 -0.9779308 … -0.98363125 -0.98363125; 0.99430096 0.7648038 … 0.7449523 0.7449523]

Stacktrace:
 [1] macro expansion
   @ ~/hostedtoolcache/julia/1.11.3/x64/share/julia/stdlib/v1.11/Test/src/Test.jl:679 [inlined]
 [2] macro expansion
   @ ~/work/Reactant.jl/Reactant.jl/test/sharding.jl:97 [inlined]
 [3] macro expansion
   @ ~/hostedtoolcache/julia/1.11.3/x64/share/julia/stdlib/v1.11/Test/src/Test.jl:1704 [inlined]
 [4] top-level scope
   @ ~/work/Reactant.jl/Reactant.jl/test/sharding.jl:60

This was failing already in #743, which was merged with failing tests.