Use AVX512 scatter/gather
AVX512 has _mm512_i32gather_ps
/_mm512_i32gather_pd
that can load strided data into a vector and _mm512_i32scatter_ps
/_mm512_i32scatter_pd
which can write a vector to strided memory. These can be used to enable vectorization without assume_inner_stride_one=True
. For LBM, this would permit vectorizing the streaming and collision kernels simultaneously. In general, it would also allow vectorizing with zyxf memory layout. I don't know how big the performance benefits would be, but it's certainly worth trying.
The Fujitsu A64fx also has special hardware support for scatter/gather with SVE instructions and would probably perform even better.
Edited by Michael Kuron