mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 21:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
26,443 Commits 506 Branches 263 Tags
rocm-main
Commit Graph

3 Commits

Author SHA1 Message Date
Nitin Srinivasan
0ed42dcdd0 Print the list of installed packages before running pytests 
Also, do not upgrade packages and disable editable mode when installing JAX at head

PiperOrigin-RevId: 732208266
 2025-02-28 11:46:12 -08:00
Nitin Srinivasan
5ae0e58a4a Update the calculation for num_processes and num_test_jobs that are used in CUDA test jobs 
We need to set them as `min(num_cpu_cores, num_gpus * max_tests_per_gpu, total ram in GB/6)` where max_tests_per_gpu = (GPU memory / 2GB)

PiperOrigin-RevId: 731730857
 2025-02-27 08:11:51 -08:00
Nitin Srinivasan
c78487d23d Add Github action workflows for running continuous tests with Pytest 
Changes:
- Adds `wheel_tests.yml` that will be used to run continuous jobs that builds artifacts and runs CPU/CUDA tests. Jobs will run by workflow calls to `build_artifacts.yml`/`pytest_cpu.yml`/`pytest_gpu.yml`.
- Adds testing of CUDA tests on H100 gpus
- Make script executable
- Change the name of GPU scripts and workflows to CUDA to be more clear as to what is being tested
PiperOrigin-RevId: 715500412
 2025-01-14 13:10:51 -08:00