Skip to content

Commit e2369d3

Browse files
authored
Fix experimental CI (#2005)
* up * up
1 parent 9a9ecde commit e2369d3

File tree

2 files changed

+6
-8
lines changed

2 files changed

+6
-8
lines changed

.github/workflows/torchao_experimental_test.yml

+3-8
Original file line numberDiff line numberDiff line change
@@ -37,9 +37,7 @@ jobs:
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: |
@@ -99,11 +97,8 @@ jobs:
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

dev-requirements.txt

+3
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,9 @@ importlib_metadata
2626
# Custom CUDA Extensions
2727
ninja
2828

29+
# CPU kernels
30+
cmake<4.0.0,>=3.19.0
31+
2932
# Linting
3033
ruff==0.6.8
3134
pre-commit

0 commit comments

Comments
 (0)