`error: 'stablehlo.concatenate' op failed to infer returned types`

```
loc(callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/LLVM/b3kFs/src/interop/base.jl" in ".">, name = "macro expansion;", linkageName = "macro expansion", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/LLVM/b3kFs/src/interop/base.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/LLVM/b3kFs/src/interop/base.jl":39:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"none" in ".">, name = "macro expansion;", linkageName = "macro expansion", file = <"none" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["none":0:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"none" in ".">, name = "pointerset;", linkageName = "pointerset", file = <"none" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["none":0:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/LLVM/b3kFs/src/interop/pointer.jl" in ".">, name = "unsafe_store!;", linkageName = "unsafe_store!", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/LLVM/b3kFs/src/interop/pointer.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/LLVM/b3kFs/src/interop/pointer.jl":88:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl" in ".">, name = "arrayset_bits;", linkageName = "arrayset_bits", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl":271:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl" in ".">, name = "#arrayset;", linkageName = "#arrayset", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl":264:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl" in ".">, name = "setindex!;", linkageName = "setindex!", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl":306:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl" in ".">, name = "setindex!;", linkageName = "setindex!", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl":318:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/OffsetArrays/b7qpm/src/OffsetArrays.jl" in ".">, name = "setindex!;", linkageName = "setindex!", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/OffsetArrays/b7qpm/src/OffsetArrays.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/OffsetArrays/b7qpm/src/OffsetArrays.jl":442:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions_flux.jl" in ".">, name = "_fill_flux_south_halo!;", linkageName = "_fill_flux_south_halo!", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions_flux.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions_flux.jl":17:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions_flux.jl" in ".">, name = "_fill_south_halo!;", linkageName = "_fill_south_halo!", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions_flux.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions_flux.jl":30:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions.jl" in ".">, name = "#34;", linkageName = "#34", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions.jl":280:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"ntuple.jl" in ".">, name = "ntuple;", linkageName = "ntuple", file = <"ntuple.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["ntuple.jl":50:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/KernelAbstractions/sWSE0/src/macros.jl" in ".">, name = "gpu__fill_south_and_north_halo!;", linkageName = "gpu__fill_south_and_north_halo!", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/KernelAbstractions/sWSE0/src/macros.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/KernelAbstractions/sWSE0/src/macros.jl":322:0] at fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, name = "gpu__fill_south_and_north_halo!", linkageName = "julia_gpu__fill_south_and_north_halo!_87315", file = <"none" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["none":0:0]))))))))))))))): error: 'stablehlo.concatenate' op inferred type(s) 'tensor<130x8163x4080xf64>' are incompatible with return type(s) of operation 'tensor<130x4082x4080xf64>'
loc(callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/LLVM/b3kFs/src/interop/base.jl" in ".">, name = "macro expansion;", linkageName = "macro expansion", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/LLVM/b3kFs/src/interop/base.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/LLVM/b3kFs/src/interop/base.jl":39:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"none" in ".">, name = "macro expansion;", linkageName = "macro expansion", file = <"none" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["none":0:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"none" in ".">, name = "pointerset;", linkageName = "pointerset", file = <"none" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["none":0:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/LLVM/b3kFs/src/interop/pointer.jl" in ".">, name = "unsafe_store!;", linkageName = "unsafe_store!", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/LLVM/b3kFs/src/interop/pointer.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/LLVM/b3kFs/src/interop/pointer.jl":88:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl" in ".">, name = "arrayset_bits;", linkageName = "arrayset_bits", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl":271:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl" in ".">, name = "#arrayset;", linkageName = "#arrayset", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl":264:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl" in ".">, name = "setindex!;", linkageName = "setindex!", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl":306:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl" in ".">, name = "setindex!;", linkageName = "setindex!", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/ext/ReactantCUDAExt.jl":318:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/OffsetArrays/b7qpm/src/OffsetArrays.jl" in ".">, name = "setindex!;", linkageName = "setindex!", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/OffsetArrays/b7qpm/src/OffsetArrays.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/OffsetArrays/b7qpm/src/OffsetArrays.jl":442:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions_flux.jl" in ".">, name = "_fill_flux_south_halo!;", linkageName = "_fill_flux_south_halo!", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions_flux.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions_flux.jl":17:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions_flux.jl" in ".">, name = "_fill_south_halo!;", linkageName = "_fill_south_halo!", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions_flux.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions_flux.jl":30:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions.jl" in ".">, name = "#34;", linkageName = "#34", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/Oceananigans/L7qIm/src/BoundaryConditions/fill_halo_regions.jl":280:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"ntuple.jl" in ".">, name = "ntuple;", linkageName = "ntuple", file = <"ntuple.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["ntuple.jl":50:0] at callsite(fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, scope = #llvm.di_file<"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/KernelAbstractions/sWSE0/src/macros.jl" in ".">, name = "gpu__fill_south_and_north_halo!;", linkageName = "gpu__fill_south_and_north_halo!", file = <"/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/KernelAbstractions/sWSE0/src/macros.jl" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["/capstor/scratch/cscs/mgiordan/santis/juliaup/depot/packages/KernelAbstractions/sWSE0/src/macros.jl":322:0] at fused<#llvm.di_subprogram, compileUnit = , sourceLanguage = DW_LANG_Julia, file = <"julia" in ".">, producer = "julia", isOptimized = true, emissionKind = None, nameTableKind = None>, name = "gpu__fill_south_and_north_halo!", linkageName = "julia_gpu__fill_south_and_north_halo!_87315", file = <"none" in ".">, subprogramFlags = "Definition|Optimized", type = <>>>["none":0:0]))))))))))))))): error: 'stablehlo.concatenate' op failed to infer returned types
┌ Error: Compilation failed, MLIR module written to /capstor/scratch/cscs/mgiordan/santis/repo/GB-25/sharding/runs/2025-04-16T01-08-42.355_NrfG/mlir_dumps/732990.0/module_189_s7kB_post_to_mhlo_shardings_pm.mlir
└ @ Reactant.MLIR.IR /capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/src/mlir/IR/Pass.jl:116
ERROR: LoadError: "failed to run pass manager on module"
Stacktrace:
[1] run!(pm::Reactant.MLIR.IR.PassManager, mod::Reactant.MLIR.IR.Module, key::String)
@ Reactant.MLIR.IR /capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/src/mlir/IR/Pass.jl:160
[2] run_pass_pipeline!(mod::Reactant.MLIR.IR.Module, pass_pipeline::String, key::String; enable_verifier::Bool)
@ Reactant.Compiler /capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/src/Compiler.jl:898
[3] run_pass_pipeline!
@ /capstor/scratch/cscs/mgiordan/santis/juliaup/depot/dev/Reactant/src/Compiler.jl:893 [inlined]
```

https://gist.github.com/giordano/7a1eb086fbf0f9d084d4c7bdac318302