Skip to content

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

.github/workflows/e2e-reusable.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ jobs:
135135

136136
- name: Install python test dependencies
137137
run: |
138-
pip install pyyaml pandas scipy numpy psutil pyre_extensions torchrec
138+
pip install pyyaml pandas scipy 'numpy==1.26.4' psutil pyre_extensions torchrec
139139
140140
- name: Install transformers package
141141
if: ${{ inputs.suite == 'huggingface' }}

.github/workflows/e2e-windows.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ jobs:
165165
- name: Install python test dependencies
166166
run: |
167167
.venv\Scripts\activate.ps1
168-
pip install pyyaml pandas scipy numpy psutil pyre_extensions torchrec
168+
pip install pyyaml pandas scipy 'numpy==1.26.4' psutil pyre_extensions torchrec
169169
170170
- name: Install transformers package
171171
if: inputs.suite == 'all' || inputs.suite == 'huggingface'

scripts/test-triton.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -468,7 +468,7 @@ run_inductor_tests() {
468468
git checkout $rev
469469
)
470470

471-
pip install pyyaml pandas scipy numpy psutil pyre_extensions torchrec
471+
pip install pyyaml pandas scipy 'numpy==1.26.4' psutil pyre_extensions torchrec
472472

473473
# TODO: Find the fastest Hugging Face model
474474
ZE_AFFINITY_MASK=0 python pytorch/benchmarks/dynamo/huggingface.py --accuracy --float32 -dxpu -n10 --no-skip --dashboard --inference --freezing --total-partitions 1 --partition-id 0 --only AlbertForMaskedLM --backend=inductor --timeout=4800 --output=$(pwd -P)/inductor_log.csv

0 commit comments

Comments
 (0)