Fix SIMD store index

7huahua · May 29, 2020 · f6b9931 · f6b9931
1 parent ebc183e
commit f6b9931
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/04_simd/13_nbody.cpp b/04_simd/13_nbody.cpp
@@ -36,8 +36,8 @@ int main() {
       fxi = _mm256_fmadd_ps(dx, mj, fxi);
       fyi = _mm256_fmadd_ps(dy, mj, fyi);
     }
-    _mm256_store_ps(fx, fxi);
-    _mm256_store_ps(fy, fyi);
+    _mm256_store_ps(fx+i, fxi);
+    _mm256_store_ps(fy+i, fyi);
   }
   for(int i=0; i<N; i++)
     printf("%d %g %g\n",i,fx[i],fy[i]);