-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
root
committed
Aug 19, 2023
0 parents
commit 4d48889
Showing
386 changed files
with
43,217 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,34 @@ | ||
version: 2.1 | ||
|
||
# this allows you to use CircleCI's dynamic configuration feature | ||
setup: true | ||
|
||
# the path-filtering orb is required to continue a pipeline based on | ||
# the path of an updated fileset | ||
orbs: | ||
path-filtering: circleci/[email protected] | ||
|
||
workflows: | ||
# the always-run workflow is always triggered, regardless of the pipeline parameters. | ||
always-run: | ||
jobs: | ||
# the path-filtering/filter job determines which pipeline | ||
# parameters to update. | ||
- path-filtering/filter: | ||
name: check-updated-files | ||
# 3-column, whitespace-delimited mapping. One mapping per | ||
# line: | ||
# <regex path-to-test> <parameter-to-set> <value-of-pipeline-parameter> | ||
mapping: | | ||
mmrotate/.* lint_only false | ||
requirements/.* lint_only false | ||
tests/.* lint_only false | ||
tools/.* lint_only false | ||
configs/.* lint_only false | ||
.circleci/.* lint_only false | ||
base-revision: main | ||
# this is the path of the configuration we should trigger once | ||
# path filtering and pipeline parameter value updates are | ||
# complete. In this case, we are using the parent dynamic | ||
# configuration itself. | ||
config-path: .circleci/test.yml |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,11 @@ | ||
ARG PYTORCH="1.8.1" | ||
ARG CUDA="10.2" | ||
ARG CUDNN="7" | ||
|
||
FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel | ||
|
||
# To fix GPG key error when running apt-get update | ||
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub | ||
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub | ||
|
||
RUN apt-get update && apt-get install -y ninja-build libglib2.0-0 libsm6 libxrender-dev libxext6 libgl1-mesa-glx |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,19 @@ | ||
#!/bin/bash | ||
|
||
TORCH=$1 | ||
CUDA=$2 | ||
|
||
# 10.2 -> cu102 | ||
MMCV_CUDA="cu`echo ${CUDA} | tr -d '.'`" | ||
|
||
# MMCV only provides pre-compiled packages for torch 1.x.0 | ||
# which works for any subversions of torch 1.x. | ||
# We force the torch version to be 1.x.0 to ease package searching | ||
# and avoid unnecessary rebuild during MMCV's installation. | ||
TORCH_VER_ARR=(${TORCH//./ }) | ||
TORCH_VER_ARR[2]=0 | ||
printf -v MMCV_TORCH "%s." "${TORCH_VER_ARR[@]}" | ||
MMCV_TORCH=${MMCV_TORCH%?} # Remove the last dot | ||
|
||
echo "export MMCV_CUDA=${MMCV_CUDA}" >> $BASH_ENV | ||
echo "export MMCV_TORCH=${MMCV_TORCH}" >> $BASH_ENV |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,184 @@ | ||
|
||
|
||
version: 2.1 | ||
|
||
# the default pipeline parameters, which will be updated according to | ||
# the results of the path-filtering orb | ||
parameters: | ||
lint_only: | ||
type: boolean | ||
default: true | ||
|
||
jobs: | ||
lint: | ||
docker: | ||
- image: cimg/python:3.7.4 | ||
steps: | ||
- checkout | ||
- run: | ||
name: Install pre-commit hook | ||
command: | | ||
pip install pre-commit | ||
pre-commit install | ||
- run: | ||
name: Linting | ||
command: pre-commit run --all-files | ||
- run: | ||
name: Check docstring coverage | ||
command: | | ||
pip install interrogate | ||
interrogate -v --ignore-init-method --ignore-module --ignore-nested-functions --ignore-regex "__repr__" --fail-under 50 mmrotate | ||
build_cpu: | ||
parameters: | ||
# The python version must match available image tags in | ||
# https://circleci.com/developer/images/image/cimg/python | ||
python: | ||
type: string | ||
torch: | ||
type: string | ||
torchvision: | ||
type: string | ||
docker: | ||
- image: cimg/python:<< parameters.python >> | ||
resource_class: large | ||
steps: | ||
- checkout | ||
- run: | ||
name: Get MMCV_TORCH as environment variables | ||
command: | | ||
. .circleci/scripts/get_mmcv_var.sh << parameters.torch >> | ||
source $BASH_ENV | ||
- run: | ||
name: Install Libraries | ||
command: | | ||
sudo apt-get update | ||
sudo apt-get install -y ninja-build libglib2.0-0 libsm6 libxrender-dev libxext6 libgl1-mesa-glx libjpeg-dev zlib1g-dev libtinfo-dev libncurses5 libgeos-dev | ||
- run: | ||
name: Configure Python & pip | ||
command: | | ||
python -m pip install --upgrade pip | ||
python -m pip install wheel | ||
- run: | ||
name: Install PyTorch | ||
command: | | ||
python -V | ||
python -m pip install torch==<< parameters.torch >>+cpu torchvision==<< parameters.torchvision >>+cpu -f https://download.pytorch.org/whl/torch_stable.html | ||
- run: | ||
name: Install mmrotate dependencies | ||
command: | | ||
python -m pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cpu/torch${MMCV_TORCH}/index.html | ||
python -m pip install mmdet | ||
python -m pip install -r requirements.txt | ||
- run: | ||
name: Build and install | ||
command: | | ||
python -m pip install -e . | ||
- run: | ||
name: Run unittests | ||
command: | | ||
python -m coverage run --branch --source mmrotate -m pytest tests/ | ||
python -m coverage xml | ||
python -m coverage report -m | ||
build_cuda: | ||
parameters: | ||
torch: | ||
type: string | ||
cuda: | ||
type: enum | ||
enum: ["10.1", "10.2", "11.1"] | ||
cudnn: | ||
type: integer | ||
default: 7 | ||
machine: | ||
image: ubuntu-2004-cuda-11.4:202110-01 | ||
docker_layer_caching: true | ||
resource_class: gpu.nvidia.small | ||
steps: | ||
- checkout | ||
- run: | ||
name: Get MMCV_TORCH and MMCV_CUDA as environment variables | ||
command: | | ||
. .circleci/scripts/get_mmcv_var.sh << parameters.torch >> << parameters.cuda >> | ||
source $BASH_ENV | ||
- run: | ||
name: Build Docker image | ||
command: | | ||
docker build .circleci/docker -t mmrotate:gpu --build-arg PYTORCH=<< parameters.torch >> --build-arg CUDA=<< parameters.cuda >> --build-arg CUDNN=<< parameters.cudnn >> | ||
docker run --gpus all -t -d -v /home/circleci/project:/mmrotate -w /mmrotate --name mmrotate mmrotate:gpu | ||
docker exec mmrotate apt-get install -y git | ||
- run: | ||
name: Install mmrotate dependencies | ||
command: | | ||
docker exec mmrotate pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/${MMCV_CUDA}/torch${MMCV_TORCH}/index.html | ||
docker exec mmrotate pip install mmdet | ||
docker exec mmrotate pip install -r requirements.txt | ||
- run: | ||
name: Build and install | ||
command: | | ||
docker exec mmrotate pip install -e . | ||
- run: | ||
name: Run unittests | ||
command: | | ||
docker exec mmrotate python -m pytest tests/ | ||
workflows: | ||
pr_stage_lint: | ||
when: << pipeline.parameters.lint_only >> | ||
jobs: | ||
- lint: | ||
name: lint | ||
filters: | ||
branches: | ||
ignore: | ||
- main | ||
pr_stage_test: | ||
when: | ||
not: | ||
<< pipeline.parameters.lint_only >> | ||
jobs: | ||
- lint: | ||
name: lint | ||
filters: | ||
branches: | ||
ignore: | ||
- main | ||
- build_cpu: | ||
name: minimum_version_cpu | ||
torch: 1.6.0 | ||
torchvision: 0.7.0 | ||
python: 3.7.7 | ||
requires: | ||
- lint | ||
- build_cpu: | ||
name: maximum_version_cpu | ||
torch: 1.9.0 | ||
torchvision: 0.10.0 | ||
python: 3.9.0 | ||
requires: | ||
- minimum_version_cpu | ||
- hold: | ||
type: approval | ||
requires: | ||
- maximum_version_cpu | ||
- build_cuda: | ||
name: mainstream_version_gpu | ||
torch: 1.8.1 | ||
# Use double quotation mark to explicitly specify its type | ||
# as string instead of number | ||
cuda: "10.2" | ||
requires: | ||
- hold | ||
merge_stage_test: | ||
when: | ||
not: | ||
<< pipeline.parameters.lint_only >> | ||
jobs: | ||
- build_cuda: | ||
name: minimum_version_gpu | ||
torch: 1.6.0 | ||
# Use double quotation mark to explicitly specify its type | ||
# as string instead of number | ||
cuda: "10.1" | ||
filters: | ||
branches: | ||
only: | ||
- main |
Oops, something went wrong.