Skip to content

Commit

Permalink
New lightning bench (#236)
Browse files Browse the repository at this point in the history
* New lightning bench

* Add voir instrumentation

* make lightning work

* Add ForeachNode && TorchrunAllNodes

* Diffusion benchmark

* Update pins

* Improve run option logic so it is easier to set and configure after run plan instanciation

* Add prepare script

* Remove stable_baselinecompiled requirements as they are outdated

* new diffusion

* Add diffusion prepare

* Add diffusion to config

* Pin

* prevent pynvml from raising exceptions to milabench

* Add tooll to display environment variables

* Install TIMM explicitly

* Fix TIMM benchmarks

* Update opt to use stdout

* update unit tests

* update examples

---------

Co-authored-by: pierre.delaunay <[email protected]>
  • Loading branch information
Delaunay and pierre.delaunay authored Jul 24, 2024
1 parent 1c96794 commit 4affc4a
Show file tree
Hide file tree
Showing 126 changed files with 3,813 additions and 3,033 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/docs.yml
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ jobs:
steps:
- uses: actions/setup-python@v2
with:
python-version: 3.11
python-version: 3.10

- uses: actions/checkout@master
with:
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/tests_integration.yml
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ jobs:
steps:
- uses: actions/setup-python@v4
with:
python-version: '3.11'
python-version: '3.10'

- name: Check out repository code
uses: actions/checkout@v3
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/tests_run.yml
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ jobs:
- uses: conda-incubator/setup-miniconda@v2
with:
auto-activate-base: false
python-version: 3.11
python-version: "3.10"
miniconda-version: "latest"
activate-environment: test

Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/tests_unit.yml
Original file line number Diff line number Diff line change
Expand Up @@ -31,13 +31,13 @@ jobs:

- uses: actions/setup-python@v5
with:
python-version: '3.11'
python-version: '3.10'

- name: dependencies
run: |
pip install -U pip
pip install poetry
poetry env use python3.11
poetry env use python3.10
source $(poetry env info -p)/bin/activate
#
# poetry doesnot work when installing those !?
Expand Down
3 changes: 3 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -52,3 +52,6 @@ base/

benchmarks/simple
benchmarks/voir
benchmarks/*/base/
benchmarks/lightning/lightning_logs/

Loading

0 comments on commit 4affc4a

Please sign in to comment.