Skip to content

Commit

Permalink
fixup pure-python CI
Browse files Browse the repository at this point in the history
  • Loading branch information
olupton committed Aug 21, 2024
1 parent dfab493 commit 0161916
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions .github/workflows/nsys-jax.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@ on:
push:
branches:
- main
- olupton/tmp

env:
NSYS_JAX_PYTHON_FILES: |
Expand Down Expand Up @@ -96,6 +97,13 @@ jobs:
uses: actions/setup-python@v5
with:
python-version: '3.12'
- name: Add modern jax_nsys/ files to .zip inputs
run: |
cd .github/container/jax_nsys
for zip in ../../workflows/nsys-jax/test_data/pax_fsdp4_4proc_proc*.zip; do
zip -ur "${zip}" .
zipinfo "${zip}"
done
- name: Use nsys-jax-combine to merge profiles from multiple nsys processes
shell: bash -x -e {0}
run: |
Expand Down

0 comments on commit 0161916

Please sign in to comment.