Update

Author: 34j

src/hpc_tutorial/tutorial_dask.py

@@ -13,16 +13,16 @@
 
 def _task(seed: int) -> float:
     rng = np.random.default_rng(seed)
-    A = rng.normal(size=(10, 1000, 1000))
-    b = rng.normal(size=(10, 1000, 1))
+    A = rng.normal(size=(100, 1000, 1000))
+    b = rng.normal(size=(100, 1000, 1))
     x = np.linalg.solve(A, b)
     return x.mean()
 
 
 if __name__ == "__main__":
     with timer() as t:
         results = joblib.Parallel(n_jobs=-1)(
-            joblib.delayed(_task)(seed) for seed in range(10)
+            joblib.delayed(_task)(seed) for seed in range(2)
         )
         print(results)
     print(f"Computed results in {t.elapsed:g}s")

src/hpc_tutorial/tutorial_normal.py

@@ -4,8 +4,8 @@
 
 def _task(seed: int) -> float:
     rng = np.random.default_rng(seed)
-    A = rng.normal(size=(10, 1000, 1000))
-    b = rng.normal(size=(10, 1000, 1))
+    A = rng.normal(size=(200, 1000, 1000))
+    b = rng.normal(size=(200, 1000, 1))
     x = np.linalg.solve(A, b)
     return x.mean()