Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add GPU test runner #242

Merged
merged 27 commits into from
Feb 8, 2023
Merged

Add GPU test runner #242

Changes from 1 commit
Commits
Show all changes
27 commits
Select commit Hold shift + click to select a range
c0d07aa
Add `rstcheck` and `doc8`
michalk8 Feb 2, 2023
2b3021d
Pass `CUDA_VISIBLE_DEVICES` to `tox`
michalk8 Feb 2, 2023
6fb3e9d
Add GPU CI runner
michalk8 Feb 3, 2023
d63441d
Merge branch 'main' into feature/improve-pre-commits
michalk8 Feb 3, 2023
af3cc26
Fix Python version for GPU tests
michalk8 Feb 3, 2023
4572dd4
Try running without `tox`
michalk8 Feb 3, 2023
f6c9ea1
Fix not installing jax[cuda]
michalk8 Feb 3, 2023
6010ebd
Use different Docker image
michalk8 Feb 6, 2023
5244413
Fix escpape
michalk8 Feb 6, 2023
7aef30d
Use apt-get
michalk8 Feb 6, 2023
270b86a
Do not use `{}`
michalk8 Feb 6, 2023
68650e5
Fix not installing `git`
michalk8 Feb 6, 2023
7cb3d10
Use personal Docker image
michalk8 Feb 7, 2023
6b34346
Pin `jax[cuda]` version
michalk8 Feb 7, 2023
ab18f39
Mark grad(sqrtm) as CPU only test
michalk8 Feb 7, 2023
b682b64
Fix ICNN hessian test on GPU
michalk8 Feb 7, 2023
1236aa6
Use `eigvalsh` to check for positive-semidefinite
michalk8 Feb 7, 2023
c60cf50
Adjust tolerance in a test
michalk8 Feb 7, 2023
9991e15
Mark Sinkhorn online as CPU
michalk8 Feb 7, 2023
7c2aa8d
Run all tests on GPU
michalk8 Feb 7, 2023
2634a58
Skip more tests on GPU
michalk8 Feb 8, 2023
1fe92af
Update tolerances on k-means test
michalk8 Feb 8, 2023
669dec5
Always jit in online Sinkhorn test
michalk8 Feb 8, 2023
7dcb6b3
Use simple comparison
michalk8 Feb 8, 2023
2863be1
Only run fast GPU tests, try other GPU
michalk8 Feb 8, 2023
d6cfb78
Use previous GPU
michalk8 Feb 8, 2023
d7210f7
[ci skip] Fix test
michalk8 Feb 8, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Do not use {}
  • Loading branch information
@michalk8
michalk8 committed Feb 6, 2023
commit 270b86a7a90876bd4a500b155c10ab8c0114fe6e
2 changes: 1 addition & 1 deletion .github/workflows/tests.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ jobs:
- name: Install dependencies
run: |
apt-get update
apt-get install -y --no-install-recommends python3-{pip,dev}
apt-get install -y --no-install-recommends python3-dev python3-pip
python3 -m pip install tox

- name: Nvidia SMI
Expand Down