Partial reduction of a 3D Array to a 1D Array

[aborderies]

Hello everyone,
I would like to perform a partial reduction of a 3D IdefixArray to a 1D IdefixArray. In my case, I want to calculate the total mass present at each radius of my spherical grid. I haven't yet found a practical and efficient way of doing this using the idefix_for and idefix_reduce functions. Is there an example in the code or in the tests that performs this procedure?
Thanks in advance for your help!

[glesur]

Hi @aborderies , there is no good example of this as far as I know in the code as it is now. You can use Kokkos atomic function, but this won't be super-efficient. There are a few examples of nested reducers in the Kokkos documentation, but it's not the easiest thing to do... e.g. https://kokkos.org/kokkos-core-wiki/ProgrammingGuide/Custom-Reductions-Built-In-Reducers.html