This is a suite of tests for CUDA-enabled packages in Nixpkgs. The Nix build environment does not support GPU-access by default. So we have to do things off-the-books!
TODO:
- A simple runner script that scrapes all scripts that look like
*.test.*
and runs them. - Add support for tests in subdirectories with
shell.nix
instead of nix-shell shebang-lines. - Add test for basic PyTorch functionality
- Add test for jax Conv issue
- Add test for basic TF functionality
- Add test for NixOS/nixpkgs#163739
- Add test for NixOS/nixpkgs#153542 (comment)
TODO: there will be a runner script (coming soon!)
Create a new script in the tests/
subdirectory with .test.
in the filename.
See jax.test.py for an example.