based-robotics · Dec 7, 2024
diff --git a/‎benchmarks/cuda_benchmark_results.npz
-4.69 KB b/‎benchmarks/cuda_benchmark_results.npz
-4.69 KB
diff --git a/‎benchmarks/jaxadi_benchmark_results.npz
-4.69 KB b/‎benchmarks/jaxadi_benchmark_results.npz
-4.69 KB
diff --git a/‎benchmarks/run_benchmark.py
+4-4 b/‎benchmarks/run_benchmark.py
+4-4
diff --git a/‎benchmarks/visualize.ipynb
+13-12 b/‎benchmarks/visualize.ipynb
+13-12
diff --git a/‎docs/static/images/compare_1e1_bar.png
9.22 KB b/‎docs/static/images/compare_1e1_bar.png
9.22 KB
diff --git a/‎docs/static/images/compare_1e2_bar.png
3.19 KB b/‎docs/static/images/compare_1e2_bar.png
3.19 KB
diff --git a/‎docs/static/images/compare_1e3_bar.png
32.1 KB b/‎docs/static/images/compare_1e3_bar.png
32.1 KB
diff --git a/‎docs/static/images/compare_1e4_bar.png
29.6 KB b/‎docs/static/images/compare_1e4_bar.png
29.6 KB
diff --git a/‎docs/static/images/speedup_ratio.png
19.4 KB b/‎docs/static/images/speedup_ratio.png
19.4 KB
@@ -17,7 +17,7 @@
 class Paths:
     FUNCTIONS_DIR = os.path.join(cur_dir, "data")
     OUTPUT_DIR = cur_dir
-    RUN_CUSADI = False
+    RUN_CUSADI = True
 
 
 class ColabPaths:
@@ -43,11 +43,11 @@ class ColabPaths:
 
     torch.manual_seed(0)
 
-N_ENVS_SWEEP = [2**i for i in range(1)]
-N_EVALS = 20
+N_ENVS_SWEEP = [2**i for i in range(15)]
+N_EVALS = 10
 
 # Load functions for CUDA benchmarking
-fn_files = ["fn_1e1.casadi", "fn_1e2.casadi", "fn_1e3.casadi", "fn_1e4.casadi", "fn_1e5.casadi"]
+fn_files = ["fn_1e1.casadi", "fn_1e2.casadi", "fn_1e3.casadi", "fn_1e4.casadi"]
 benchmark_fns = [Function.load(os.path.join(PathsProvider.FUNCTIONS_DIR, fn)) for fn in fn_files]