Create sharded_ocean_climate_simulation_run.jl

[glwagner]

No description provided.

[giordano]

https://github.com/PRONTOLab/GB-25/actions/runs/14296035144/job/40063319842?pr=150#step:10:146

ERROR: LoadError: TypeError: in typeassert, expected ConcreteIFRTNumber{Float64, Reactant.Sharding.ShardInfo{Reactant.Sharding.NamedSharding{0, Reactant.Sharding.Mesh{3, Vector{Int64}}}, Vector{Tuple{}}}}, got a value of type ConcreteIFRTNumber{Float64, Reactant.Sharding.ShardInfo{Reactant.Sharding.Replicated{Reactant.Sharding.Mesh{1, UnitRange{Int64}}}, Vector{Tuple{}}}}
Stacktrace:
 [1] deepcopy_internal(x::Any, stackdict::IdDict{Any, Any})
   @ Base ./deepcopy.jl:69
 [2] deepcopy
   @ ./deepcopy.jl:29 [inlined]
 [3] OceanSeaIceModel
   @ ~/.julia/packages/ClimaOcean/UuK0U/src/OceanSeaIceModels/ocean_sea_ice_model.jl:108 [inlined]
 [4] macro expansion
   @ ~/work/GB-25/GB-25/src/model_utils.jl:36 [inlined]
 [5] data_free_ocean_climate_model_init(arch::Distributed{ReactantState, nothing, Partition{Int64, Int64, Int64}, Tuple{Int64, Int64, Int64}, Int64, Tuple{Int64, Int64, Int64}, Reactant.Sharding.Mesh{3, Vector{Int64}}, Nothing, Nothing, Base.RefValue{Int64}, Vector{Reactant.XLA.IFRT.Device}}, Nx::Int64, Ny::Int64, Nz::Int64; halo::Tuple{Int64, Int64, Int64}, Δt::Int64, grid_type::Symbol)
   @ GordonBell25 ~/work/GB-25/GB-25/src/data_free_ocean_climate_model.jl:74
 [6] top-level scope
   @ ~/work/GB-25/GB-25/sharding/sharded_ocean_climate_simulation_run.jl:55
in expression starting at /home/runner/work/GB-25/GB-25/sharding/sharded_ocean_climate_simulation_run.jl:55

CC @avik-pal. Sharding of the number is incorrect?

[avik-pal]

I think I can fix this.

[avik-pal]

try EnzymeAD/Reactant.jl#1149

[giordano]

https://github.com/PRONTOLab/GB-25/actions/runs/14296946311/job/40065314773?pr=150#step:10:26

ERROR: LoadError: UndefKeywordError: keyword argument `resolution` not assigned
Stacktrace:
 [1] data_free_ocean_climate_model_init(arch::ReactantState)
   @ GordonBell25 ~/work/GB-25/GB-25/src/data_free_ocean_climate_model.jl:12
 [2] top-level scope
   @ ~/work/GB-25/GB-25/simulations/ocean_climate_simulation_run.jl:13
in expression starting at /home/runner/work/GB-25/GB-25/simulations/ocean_climate_simulation_run.jl:13

@glwagner this is on you 😁