Skip to content

Commit

Permalink
Merge pull request #455 from mlcommons/revert-453-docker_automation_fix
Browse files Browse the repository at this point in the history
Revert "Docker automation fix"
  • Loading branch information
priyakasimbeg authored Jul 26, 2023
2 parents 153901e + e290457 commit db37df4
Show file tree
Hide file tree
Showing 56 changed files with 1,702 additions and 6,669 deletions.
90 changes: 24 additions & 66 deletions .github/workflows/CI.yml
Original file line number Diff line number Diff line change
Expand Up @@ -6,13 +6,11 @@ jobs:
fastmri:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- uses: actions/checkout@v2
- name: Set up Python 3.9
uses: actions/setup-python@v4
uses: actions/setup-python@v2
with:
python-version: 3.9
cache: 'pip' # Cache pip dependencies\.
cache-dependency-path: '**/setup.py'
- name: Install Modules and Run
run: |
pip install .[jax_cpu]
Expand All @@ -24,13 +22,11 @@ jobs:
wmt_jax:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- uses: actions/checkout@v2
- name: Set up Python 3.9
uses: actions/setup-python@v4
uses: actions/setup-python@v2
with:
python-version: 3.9
cache: 'pip' # Cache pip dependencies\.
cache-dependency-path: '**/setup.py'
- name: Install Modules and Run
run: |
pip install .[jax_cpu]
Expand All @@ -41,13 +37,11 @@ jobs:
wmt_pytorch:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- uses: actions/checkout@v2
- name: Set up Python 3.9
uses: actions/setup-python@v4
uses: actions/setup-python@v2
with:
python-version: 3.9
cache: 'pip' # Cache pip dependencies\.
cache-dependency-path: '**/setup.py'
- name: Install Modules and Run
run: |
pip install .[jax_cpu]
Expand All @@ -58,13 +52,11 @@ jobs:
imagenet_jax:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- uses: actions/checkout@v2
- name: Set up Python 3.9
uses: actions/setup-python@v4
uses: actions/setup-python@v2
with:
python-version: 3.9
cache: 'pip' # Cache pip dependencies\.
cache-dependency-path: '**/setup.py'
- name: Install Modules and Run
run: |
pip install .[jax_cpu]
Expand All @@ -76,13 +68,11 @@ jobs:
imagenet_pytorch:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- uses: actions/checkout@v2
- name: Set up Python 3.9
uses: actions/setup-python@v4
uses: actions/setup-python@v2
with:
python-version: 3.9
cache: 'pip' # Cache pip dependencies\.
cache-dependency-path: '**/setup.py'
- name: Install Modules and Run
run: |
pip install .[jax_cpu]
Expand All @@ -95,13 +85,11 @@ jobs:
criteo_jax:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- uses: actions/checkout@v2
- name: Set up Python 3.9
uses: actions/setup-python@v4
uses: actions/setup-python@v2
with:
python-version: 3.9
cache: 'pip' # Cache pip dependencies\.
cache-dependency-path: '**/setup.py'
- name: Install Modules and Run
run: |
pip install .[jax_cpu]
Expand All @@ -112,13 +100,11 @@ jobs:
criteo_pytorch:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- uses: actions/checkout@v2
- name: Set up Python 3.9
uses: actions/setup-python@v4
uses: actions/setup-python@v2
with:
python-version: 3.9
cache: 'pip' # Cache pip dependencies\.
cache-dependency-path: '**/setup.py'
- name: Install Modules and Run
run: |
pip install .[jax_cpu]
Expand All @@ -129,13 +115,11 @@ jobs:
speech_jax:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- uses: actions/checkout@v2
- name: Set up Python 3.9
uses: actions/setup-python@v4
uses: actions/setup-python@v2
with:
python-version: 3.9
cache: 'pip' # Cache pip dependencies\.
cache-dependency-path: '**/setup.py'
- name: Install Modules and Run
run: |
pip install .[jax_cpu]
Expand All @@ -147,13 +131,11 @@ jobs:
speech_pytorch:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- uses: actions/checkout@v2
- name: Set up Python 3.9
uses: actions/setup-python@v4
uses: actions/setup-python@v2
with:
python-version: 3.9
cache: 'pip' # Cache pip dependencies\.
cache-dependency-path: '**/setup.py'
- name: Install Modules and Run
run: |
pip install .[jax_cpu]
Expand All @@ -165,13 +147,11 @@ jobs:
ogbg:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- uses: actions/checkout@v2
- name: Set up Python 3.9
uses: actions/setup-python@v4
uses: actions/setup-python@v2
with:
python-version: 3.9
cache: 'pip' # Cache pip dependencies\.
cache-dependency-path: '**/setup.py'
- name: Install Modules and Run
run: |
pip install .[jax_cpu]
Expand All @@ -183,44 +163,22 @@ jobs:
pytest:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- uses: actions/checkout@v2
- name: Set up Python 3.9
uses: actions/setup-python@v4
uses: actions/setup-python@v2
with:
python-version: 3.9
cache: 'pip' # Cache pip dependencies\.
cache-dependency-path: '**/setup.py'
- name: Install pytest
run: |
python -m pip install --upgrade pip
pip install pytest
pip install .[full]
pip install .[jax_cpu]
pip install .[pytorch_cpu]
- name: Run pytest tests
- name: Run pytest
run: |
pytest -vx tests/version_test.py
pytest -vx tests/test_num_params.py
pytest -vx tests/test_param_shapes.py
pytest -vx tests/test_param_types.py
pytest -vx tests/test_ssim.py
pytest-baselines:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- name: Set up Python 3.9
uses: actions/setup-python@v4
with:
python-version: 3.9
cache: 'pip' # Cache pip dependencies\.
cache-dependency-path: '**/setup.py'
- name: Install pytest
run: |
python -m pip install --upgrade pip
pip install pytest
pip install .[full]
pip install .[jax_cpu]
pip install .[pytorch_cpu]
- name: Run baseline tests
run: |
pytest --verbosity=1 tests/test_baselines.py
pytest -vx tests/test_ssim.py
136 changes: 0 additions & 136 deletions .github/workflows/regression_tests.yml

This file was deleted.

Loading

0 comments on commit db37df4

Please sign in to comment.