Skip to content

Commit 16ce746

Browse files
authored
Use numpy 1.26 in CI
Numpy installed with pip install causes some issues for pytorch. 1.26 is stable.
1 parent 80626a9 commit 16ce746

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

.github/workflows/amd_perf_kernel_Integration_tests.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,7 @@ jobs:
123123
pip uninstall -y triton
124124
cd python
125125
pip install -v -e .
126+
pip install numpy==1.26.4
126127
- name: Run Perf Kernels Unit Tests
127128
run: |
128129
pytest -vvv ./python/perf-kernels/flash-attention.py

.github/workflows/amd_perf_kernel_postmerge_tests.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,7 @@ jobs:
8484
pip uninstall -y triton
8585
cd python
8686
pip install -v -e .
87+
pip install numpy==1.26.4
8788
- name: Run Perf Kernels Unit Tests
8889
run: |
8990
pytest -vvv ./python/perf-kernels/flash-attention.py

0 commit comments

Comments
 (0)