Skip to content

Commit ac301f5

Browse files
authored
Merge pull request #30 from graphcore-research/test-ci
Fix torch/numpy versions for MX in tests
2 parents 5e409f8 + c4e4c72 commit ac301f5

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

requirements-test.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,6 @@
11
# PyPI doesn't like direct dependencies - see https://github.com/microsoft/microxcaling/issues/22
22

3+
numpy<2.0.0 # microscaling needs pytorch 2.1, which needs to be less than 2.0
4+
torch==2.1.0 --index-url https://download.pytorch.org/whl/cpu
5+
36
mx @ git+https://github.com/microsoft/microxcaling

0 commit comments

Comments
 (0)