Skip to content

Commit

Permalink
Install torch
Browse files Browse the repository at this point in the history
  • Loading branch information
@MichaelHudgins
MichaelHudgins committed Oct 22, 2024
1 parent bd3e4cc commit b0fba6b
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions .github/workflows/ci.yml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,7 @@ jobs:
python-version: '3.10'
- name: Setup Released JAX
run: |
pip install torch
pip install -U "jax[cuda12]"
pip install pytest
- name: Test JAX Triton
Expand Down

0 comments on commit b0fba6b

Please sign in to comment.