fix: update plots for graph translation

based-robotics · Dec 7, 2024 · 64162de · 64162de
1 parent f81465c
commit 64162de
Show file tree

Hide file tree

Showing 9 changed files with 17 additions and 16 deletions.
diff --git a/benchmarks/cuda_benchmark_results.npz b/benchmarks/cuda_benchmark_results.npz
diff --git a/benchmarks/jaxadi_benchmark_results.npz b/benchmarks/jaxadi_benchmark_results.npz
diff --git a/benchmarks/run_benchmark.py b/benchmarks/run_benchmark.py
@@ -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]
 
 

diff --git a/benchmarks/visualize.ipynb b/benchmarks/visualize.ipynb
diff --git a/docs/static/images/compare_1e1_bar.png b/docs/static/images/compare_1e1_bar.png
diff --git a/docs/static/images/compare_1e2_bar.png b/docs/static/images/compare_1e2_bar.png
diff --git a/docs/static/images/compare_1e3_bar.png b/docs/static/images/compare_1e3_bar.png
diff --git a/docs/static/images/compare_1e4_bar.png b/docs/static/images/compare_1e4_bar.png
diff --git a/docs/static/images/speedup_ratio.png b/docs/static/images/speedup_ratio.png