Fix wrong simd tag in AVX all_reduce