Skip to content

Commit

Permalink
Added circleci job for testing cuda 11 wheel
Browse files Browse the repository at this point in the history
  • Loading branch information
luisenp committed Nov 7, 2022
1 parent 3bbbb15 commit 10df838
Showing 1 changed file with 24 additions and 0 deletions.
24 changes: 24 additions & 0 deletions .circleci/config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -176,6 +176,16 @@ run_tests: &run_tests
pytest -s tests/test_theseus_layer.py
pytest -s tests -m "cudaext"
build_cuda11_wheel: &build_cuda11_wheel
- run:
name: Building wheel for CUDA 11
working_directory: /home/circleci
command: |
THESEUS_GIT_COMMIT=$(git --git-dir project/.git log --format="%H" -n 1)
THESEUS_VERSION=$(grep -Eo "[0-9].[0-9].[0-9][.0-9a-z]*" project/theseus/__init__.py)
./project/build_scripts/build_wheel.sh . ${THESEUS_GIT_COMMIT} 11.6 ${THESEUS_VERSION}
# -------------------------------------------------------------------------------------
# Jobs
# -------------------------------------------------------------------------------------
Expand Down Expand Up @@ -258,6 +268,19 @@ jobs:
- <<: *setup_project
- <<: *run_tests

test_cuda11_wheel:
docker:
- image: cimg/python:3.9
steps:
- checkout
- <<: *update_and_install_python
- <<: *setup_cuda11_libs
- <<: *setup_environment
- <<: *install_torch_cuda11
- <<: *build_cuda11_wheel



workflows:
version: 2
build:
Expand All @@ -267,3 +290,4 @@ workflows:
- py39_linux
- unittests_gpu17_cuda10
- unittests_gpu17_cuda11
- test_cuda11_wheel

0 comments on commit 10df838

Please sign in to comment.