update codes and add updated plot.

Author: khushi-411

benchmarks/python/acc_pythran.py

@@ -26,30 +26,22 @@ def compute_accelerations(accelerations, masses, positions):
         position0 = positions[index_p0]
         mass0 = masses[index_p0]
 
-        vectors = position0 - positions[index_p0 + 1: nb_particles]
-
-        distances = np.square(vectors).sum(axis = 1)
-        coefs = distances ** 1.5
-
-        _x = np.add(np.sum(
-            np.divide(
-                np.multiply(
-                    masses[index_p0 + 1: nb_particles], -1 * vectors.T
-                    ),
-                coefs).T, axis = 0),
-            accelerations[index_p0]
-            )
-        accelerations[index_p0] = _x
-
-        _temp = np.add(
-                np.divide(
-                np.multiply(
-                    mass0, vectors.T
-                    ),
-                coefs).T,
-            accelerations[index_p0 + 1: nb_particles]
-            )
-        accelerations[index_p0 + 1: nb_particles] = _temp
+        vector = np.empty(3)
+
+        for index_p1 in range(index_p0 + 1, nb_particles):
+            mass1 = masses[index_p1]
+            position1 = positions[index_p1]
+
+            for i in range(3):
+                vector[i] = position0[i] - position1[i]
+
+            distance = sum(vector ** 2) * math.sqrt(sum(vector ** 2))
+            coef_m1 = mass0 / distance
+            coef_m2 = mass1 / distance
+
+            for i in range(3):
+                accelerations[index_p0][i] += coef_m1 * -1 * vector[i]
+                accelerations[index_p1][i] += coef_m2 * vector[i]
 
     return accelerations
 

benchmarks/python/plot.py

@@ -13,14 +13,14 @@ def plot(x, labels, list_df, names):
 
     list1 = []
     for ind, list in enumerate(list_df):
-        if ind < 3:
+        if ind < 2:
             list1.append(list)
 
     for ind, time_taken in enumerate(list1):
         rect[ind] = plt.bar(x + width * ind, time_taken, width, align = 'center', label = labels[ind])
         plt.bar_label(rect[ind], padding = 3, fontsize = 250)
 
-    plt.xticks(x + width, names, fontsize = 300)
+    plt.xticks(x + width / 2, names, fontsize = 300)
     plt.yticks(fontsize = 300)
     plt.legend(fontsize = 250)
 
@@ -34,10 +34,10 @@ def plot(x, labels, list_df, names):
 
     list2 = []
     for ind, list in enumerate(list_df):
-        if ind >= 3:
+        if ind >= 2:
             list2.append(list)
 
-    j = 3
+    j = 2
     for ind, time_taken in enumerate(list2):
         rect[ind] = plt.bar(x + width * ind, time_taken, width, align = 'center', color = colors[ind % len(colors)], label = labels[j])
         j += 1
@@ -55,10 +55,11 @@ def plot(x, labels, list_df, names):
     plt.savefig("benchmarking-numpy")
 
 if __name__ == "__main__":
-    df = pd.read_csv("data.csv")
+    data_path = "table.csv"
+    df = pd.read_csv(data_path)
     df = df.drop(['Unnamed: 0'], axis = 1)
     df = df.T
-    df.columns = ["Python-NumPy", "Pure-NumPy", "Pure-Python", "C++", "Numba", "Pythran: Transonic"]
+    df.columns = ["NumPy", "Python", "C++", "Numba", "Pythran: Transonic"]
 
     labels = df.columns
     names = ["16", "32", "64", "128"]

content/en/benchmark.md

@@ -183,10 +183,10 @@ table, th, td {
  </tr>
  <tr>
   <td><b>Pythran-Naive: Transonic</b></td>
-  <td>0.0023</td>
-  <td>0.0011</td>
-  <td>0.0004</td>
-  <td>0.0002</td>
+  <td>0.0111</td>
+  <td>0.0089</td>
+  <td>0.0083</td>
+  <td>0.0078</td>
  </tr>
 </table>
 </body>