1- name : test-optimizers
2- on :
3- pull_request :
4- branches :
5- - main
1+ # name: test-optimizers
2+ # on:
3+ # pull_request:
4+ # branches:
5+ # - main
66
7- jobs :
8- test_optimizers :
9- runs-on : ubuntu-latest
10- strategy :
11- matrix :
12- python-version : [3.8, 3.9, 3.10.15]
13- steps :
14- 15- - name : Set up Python ${{ matrix.python-version }}
16- 17- with :
18- python-version : ${{ matrix.python-version }}
19- 20- - name : Install suitesparse
21- run : |
22- sudo apt-get update && sudo apt-get install -y libsuitesparse-dev
23- - name : Create Conda env
24- run : |
25- wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh
26- /bin/bash ~/miniconda.sh -b -p ~/conda
27- export PATH=~/conda/bin:$PATH
28- conda create --name theseus python=${{ matrix.python-version }}
29- source activate theseus
30- pip install --progress-bar off --upgrade pip
31- pip install --progress-bar off --upgrade setuptools
32- - name : Install Torch
33- run : |
34- export PATH=~/conda/bin:$PATH
35- source activate theseus
36- pip install torch
37- - name : Build Baspacho CPU
38- run : |
39- sudo apt-get install -y libopenblas-pthread-dev
40- git clone https://github.com/facebookresearch/baspacho.git
41- cd baspacho
42- echo $(pwd)
43- cmake -S . -B build -DCMAKE_BUILD_TYPE=Release -DBLA_STATIC=ON -DBASPACHO_USE_CUBLAS=0 -DBUILD_SHARED_LIBS=OFF -DBASPACHO_BUILD_TESTS=OFF -DBASPACHO_BUILD_EXAMPLES=OFF
44- cmake --build build -- -j16
45- - name : Install torchlie and torchkin
46- run : |
47- export PATH=~/conda/bin:$PATH
48- source activate theseus
49- cd torchlie
50- pip install -e .
51- cd ../torchkin
52- pip install -e .
53- - name : Install theseus
54- run : |
55- export PATH=~/conda/bin:$PATH
56- source activate theseus
57- echo $(pwd)
58- BASPACHO_ROOT_DIR=./baspacho pip install -e ".[dev]"
59- - name : Run Optimizer tests
60- run : |
61- export PATH=~/conda/bin:$PATH
62- source activate theseus
63- # the full baspacho test for CPU fails, we only run an abriged version in CI
64- python -m pytest tests/theseus_tests/optimizer -m "not cudaext" -k "not test_baspacho_solver_cpu_full"
65- python -m pytest tests/theseus_tests/test_theseus_layer.py -m "not cudaext"
7+ # jobs:
8+ # test_optimizers:
9+ # runs-on: ubuntu-latest
10+ # strategy:
11+ # matrix:
12+ # python-version: [3.8, 3.9, 3.10.15]
13+ # steps:
14+ # - uses: actions/[email protected] 15+ # - name: Set up Python ${{ matrix.python-version }}
16+ 17+ # with:
18+ # python-version: ${{ matrix.python-version }}
19+ # - uses: actions/[email protected] 20+ # - name: Install suitesparse
21+ # run: |
22+ # sudo apt-get update && sudo apt-get install -y libsuitesparse-dev
23+ # - name: Create Conda env
24+ # run: |
25+ # wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh
26+ # /bin/bash ~/miniconda.sh -b -p ~/conda
27+ # export PATH=~/conda/bin:$PATH
28+ # conda create --name theseus python=${{ matrix.python-version }}
29+ # source activate theseus
30+ # pip install --progress-bar off --upgrade pip
31+ # pip install --progress-bar off --upgrade setuptools
32+ # - name: Install Torch
33+ # run: |
34+ # export PATH=~/conda/bin:$PATH
35+ # source activate theseus
36+ # pip install torch
37+ # - name: Build Baspacho CPU
38+ # run: |
39+ # sudo apt-get install -y libopenblas-pthread-dev
40+ # git clone https://github.com/facebookresearch/baspacho.git
41+ # cd baspacho
42+ # echo $(pwd)
43+ # cmake -S . -B build -DCMAKE_BUILD_TYPE=Release -DBLA_STATIC=ON -DBASPACHO_USE_CUBLAS=0 -DBUILD_SHARED_LIBS=OFF -DBASPACHO_BUILD_TESTS=OFF -DBASPACHO_BUILD_EXAMPLES=OFF
44+ # cmake --build build -- -j16
45+ # - name: Install torchlie and torchkin
46+ # run: |
47+ # export PATH=~/conda/bin:$PATH
48+ # source activate theseus
49+ # cd torchlie
50+ # pip install -e .
51+ # cd ../torchkin
52+ # pip install -e .
53+ # - name: Install theseus
54+ # run: |
55+ # export PATH=~/conda/bin:$PATH
56+ # source activate theseus
57+ # echo $(pwd)
58+ # BASPACHO_ROOT_DIR=./baspacho pip install -e ".[dev]"
59+ # - name: Run Optimizer tests
60+ # run: |
61+ # export PATH=~/conda/bin:$PATH
62+ # source activate theseus
63+ # # the full baspacho test for CPU fails, we only run an abriged version in CI
64+ # python -m pytest tests/theseus_tests/optimizer -m "not cudaext" -k "not test_baspacho_solver_cpu_full"
65+ # python -m pytest tests/theseus_tests/test_theseus_layer.py -m "not cudaext"
0 commit comments