[bench] n_iter

Author: kngwyu

benchmark/bench_phyjax2d.py

@@ -24,11 +24,11 @@ def ball_fall_phyjax2d(
         gravity=(0.0, -900.0),
         dt=0.002,
         viscous_damping=0.6,
-        n_velocity_iter=10,
+        n_velocity_iter=4,
         n_position_iter=1,
         bias_factor=0.1,
         bounce_threshold=4,
-        allowed_penetration=0.01,
+        allowed_penetration=0.1,
     )
 
     for _ in range(n_balls):
@@ -111,12 +111,17 @@ def n_step_fixed(sd, vs):
 def main(
     counts: list[int] = DEFAULT_COUNTS,
     debug_vis: bool = False,
+    n_iter: int = 1000,
     filename: Path = Path("bench.csv"),
 ) -> None:
     results = []
 
     for count in counts:
-        duration = ball_fall_phyjax2d(count, debug_vis)
+        duration = ball_fall_phyjax2d(
+            count,
+            debug_vis,
+            n_iter=n_iter,
+        )
         # Convert timedelta to total seconds as a float for the CSV
         seconds = duration.total_seconds()
         results.append((count, seconds))

benchmark/bench_pymunk.py

@@ -9,11 +9,15 @@
 import typer
 
 
-def ball_fall(n_balls: int, debug_vis: bool, n_iter: int = 1000) -> timedelta:
+def ball_fall(
+    n_balls: int,
+    debug_vis: bool,
+    n_iter: int = 1000,
+) -> timedelta:
     space = pymunk.Space()
     # 1. Flip Gravity: Positive Y pulls "down" in PyGame coordinates
     space.gravity = (0, 900)
-    space.iterations = 10
+    space.iterations = 4
 
     static_body = space.static_body
     # 2. Invert Container: Floor is now at Y=800, walls go up toward Y=50
@@ -79,12 +83,18 @@ def ball_fall(n_balls: int, debug_vis: bool, n_iter: int = 1000) -> timedelta:
 def main(
     counts: list[int] = DEFAULT_COUNTS,
     debug_vis: bool = False,
+    n_iter: int = 1000,
+    freefall: bool = False,
     filename: Path = Path("bench.csv"),
 ) -> None:
     results = []
 
     for count in counts:
-        duration = ball_fall(count, debug_vis)
+        duration = ball_fall(
+            count,
+            debug_vis,
+            n_iter=n_iter,
+        )
         # Convert timedelta to total seconds as a float for the CSV
         seconds = duration.total_seconds()
         results.append((count, seconds))