From 44d5a36debf8c1c8516ca71dd01c268ebd604fdc Mon Sep 17 00:00:00 2001 From: BirkhoffG <26811230+BirkhoffG@users.noreply.github.com> Date: Wed, 28 Aug 2024 17:30:05 -0400 Subject: [PATCH] Update legacy benchmark scripts and dependencies --- benchmarks/README.md | 20 +++++++++++++++++++ .../{ => legacy}/benchmark_large_dataset.py | 0 .../{ => legacy}/benchmark_scalability.py | 0 3 files changed, 20 insertions(+) rename benchmarks/{ => legacy}/benchmark_large_dataset.py (100%) rename benchmarks/{ => legacy}/benchmark_scalability.py (100%) diff --git a/benchmarks/README.md b/benchmarks/README.md index 43e7368..ce3fa84 100644 --- 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 b/benchmarks/legacy/benchmark_large_dataset.py similarity index 100% rename from benchmarks/benchmark_large_dataset.py rename to benchmarks/legacy/benchmark_large_dataset.py diff --git a/benchmarks/benchmark_scalability.py b/benchmarks/legacy/benchmark_scalability.py similarity index 100% rename from benchmarks/benchmark_scalability.py rename to benchmarks/legacy/benchmark_scalability.py