Update legacy benchmark scripts and dependencies

BirkhoffG · Aug 28, 2024 · 44d5a36 · 44d5a36
1 parent 11af84e
commit 44d5a36
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 0 deletions.
diff --git a/benchmarks/README.md b/benchmarks/README.md
@@ -16,3 +16,23 @@ To reproduce the results in the paper, you can run
 ```sh
 python -m benchmarks.built-in.run_all
 ```
+
+## Legacy Scripts
+
+> [!WARNING]  
+> These scripts are used to benchmark `jax-relax<0.2.0>=0.1.0`. 
+>
+
+To run these scripts, install the dependencies as:
+```sh
+pip install "jax-relax[dev]<0.2.0"
+```
+
+Next, run the script as:
+
+```sh
+# run large dataset
+python -m benchmarks.legacy.benchmark_large_dataset
+# run scalability test
+python -m benchmarks.legacy.benchmark_scalability
+```
diff --git a/benchmarks/benchmark_large_dataset.py → benchmarks/legacy/benchmark_large_dataset.py b/benchmarks/benchmark_large_dataset.py → benchmarks/legacy/benchmark_large_dataset.py
diff --git a/benchmarks/benchmark_scalability.py → benchmarks/legacy/benchmark_scalability.py b/benchmarks/benchmark_scalability.py → benchmarks/legacy/benchmark_scalability.py