File tree Expand file tree Collapse file tree 2 files changed +6
-8
lines changed
Expand file tree Collapse file tree 2 files changed +6
-8
lines changed Original file line number Diff line number Diff line change 3737 # of torch and torchao, which we do not want to use
3838 pip install executorch
3939 pip install torch==2.7.0.dev20250311 --index-url "https://download.pytorch.org/whl/nightly/cpu" --force-reinstall
40- pip install numpy
41- pip install pytest
42- pip install parameterized
40+ pip install -r dev-requirements.txt
4341 USE_CPP=1 TOCHAO_BUILD_KLEIDIAI=1 pip install .
4442 - name : Run python tests
4543 run : |
9997 python -c "import torch; print(torch.__version__)"
10098 - name : Install requirements
10199 run : |
102- pip install cmake
103- pip install parameterized
104- pip install pyyaml
105- pip install numpy
106- pip install importlib-metadata
100+ pip install -r dev-requirements.txt
101+ pip install pyyaml importlib-metadata
107102 - name : Print pip freeze
108103 run : |
109104 pip freeze
Original file line number Diff line number Diff line change @@ -26,6 +26,9 @@ importlib_metadata
2626# Custom CUDA Extensions
2727ninja
2828
29+ # CPU kernels
30+ cmake<4.0.0,>=3.19.0
31+
2932# Linting
3033ruff==0.6.8
3134pre-commit
You can’t perform that action at this time.
0 commit comments