File tree 2 files changed +6
-8
lines changed
2 files changed +6
-8
lines changed Original file line number Diff line number Diff line change 37
37
# of torch and torchao, which we do not want to use
38
38
pip install executorch
39
39
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
43
41
USE_CPP=1 TOCHAO_BUILD_KLEIDIAI=1 pip install .
44
42
- name : Run python tests
45
43
run : |
99
97
python -c "import torch; print(torch.__version__)"
100
98
- name : Install requirements
101
99
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
107
102
- name : Print pip freeze
108
103
run : |
109
104
pip freeze
Original file line number Diff line number Diff line change @@ -26,6 +26,9 @@ importlib_metadata
26
26
# Custom CUDA Extensions
27
27
ninja
28
28
29
+ # CPU kernels
30
+ cmake<4.0.0,>=3.19.0
31
+
29
32
# Linting
30
33
ruff==0.6.8
31
34
pre-commit
You can’t perform that action at this time.
0 commit comments