mirror of https://github.com/open-mmlab/mmcv.git
Update parrots image (#2522)
* Update image * Bugfix for parrots ext ops * Move parrots ci task to build_pat.yml * Fix on:push * Remove commets and fix * Fix beautiful job namepull/2535/head
parent
859d84c5b1
commit
64e739e002
|
@ -1,6 +1,13 @@
|
||||||
name: build_pat
|
name: build_pat
|
||||||
|
|
||||||
on: push
|
on:
|
||||||
|
push:
|
||||||
|
paths-ignore:
|
||||||
|
- 'README.md'
|
||||||
|
- 'README_zh-CN.md'
|
||||||
|
- 'docs/**'
|
||||||
|
- 'examples/**'
|
||||||
|
- '.dev_scripts/**'
|
||||||
|
|
||||||
concurrency:
|
concurrency:
|
||||||
group: ${{ github.workflow }}-${{ github.ref }}
|
group: ${{ github.workflow }}-${{ github.ref }}
|
||||||
|
@ -13,14 +20,13 @@ jobs:
|
||||||
build_parrots:
|
build_parrots:
|
||||||
runs-on: ubuntu-18.04
|
runs-on: ubuntu-18.04
|
||||||
container:
|
container:
|
||||||
image: ghcr.io/zhouzaida/parrots-mmcv:1.3.4
|
image: ghcr.io/cokedong/parrots:pat0.21.0a0_cuda11
|
||||||
credentials:
|
credentials:
|
||||||
username: zhouzaida
|
username: cokedong
|
||||||
password: ${{ secrets.CR_PAT }}
|
password: ${{ secrets.PARROTS_CI_TOKEN }}
|
||||||
|
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v2
|
- uses: actions/checkout@v2
|
||||||
- name: Install unittest dependencies
|
|
||||||
run: pip install -r requirements/test.txt
|
|
||||||
- name: Build and install
|
- name: Build and install
|
||||||
run: rm -rf .eggs && MMCV_WITH_OPS=1 pip install -e .
|
run: |
|
||||||
|
source /usr/local/env/pat_latest
|
||||||
|
MMCV_WITH_OPS=1 pip install -e . -v
|
||||||
|
|
|
@ -7,6 +7,7 @@
|
||||||
|
|
||||||
using namespace parrots;
|
using namespace parrots;
|
||||||
|
|
||||||
|
#ifdef MMCV_WITH_CUDA
|
||||||
void border_align_forward_cuda_parrots(CudaContext& ctx, const SSElement& attr,
|
void border_align_forward_cuda_parrots(CudaContext& ctx, const SSElement& attr,
|
||||||
const OperatorBase::in_list_t& ins,
|
const OperatorBase::in_list_t& ins,
|
||||||
OperatorBase::out_list_t& outs) {
|
OperatorBase::out_list_t& outs) {
|
||||||
|
@ -49,3 +50,4 @@ PARROTS_EXTENSION_REGISTER(border_align_backward)
|
||||||
.output(1)
|
.output(1)
|
||||||
.apply(border_align_backward_cuda_parrots)
|
.apply(border_align_backward_cuda_parrots)
|
||||||
.done();
|
.done();
|
||||||
|
#endif
|
||||||
|
|
|
@ -40,6 +40,7 @@ void rotated_feature_align_backward_cuda_parrots(
|
||||||
rotated_feature_align_backward(grad_output, best_bboxes, grad_input,
|
rotated_feature_align_backward(grad_output, best_bboxes, grad_input,
|
||||||
spatial_scale, points);
|
spatial_scale, points);
|
||||||
}
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
void rotated_feature_align_forward_cpu_parrots(
|
void rotated_feature_align_forward_cpu_parrots(
|
||||||
HostContext& ctx, const SSElement& attr, const OperatorBase::in_list_t& ins,
|
HostContext& ctx, const SSElement& attr, const OperatorBase::in_list_t& ins,
|
||||||
|
@ -57,7 +58,6 @@ void rotated_feature_align_forward_cpu_parrots(
|
||||||
rotated_feature_align_forward(features, best_bboxes, output, spatial_scale,
|
rotated_feature_align_forward(features, best_bboxes, output, spatial_scale,
|
||||||
points);
|
points);
|
||||||
}
|
}
|
||||||
#endif
|
|
||||||
|
|
||||||
void rotated_feature_align_backward_cpu_parrots(
|
void rotated_feature_align_backward_cpu_parrots(
|
||||||
HostContext& ctx, const SSElement& attr, const OperatorBase::in_list_t& ins,
|
HostContext& ctx, const SSElement& attr, const OperatorBase::in_list_t& ins,
|
||||||
|
@ -92,7 +92,7 @@ PARROTS_EXTENSION_REGISTER(rotated_feature_align_backward)
|
||||||
.attr("points")
|
.attr("points")
|
||||||
.input(2)
|
.input(2)
|
||||||
.output(1)
|
.output(1)
|
||||||
.apply(rotated_feature_align_forward_cpu_parrots)
|
.apply(rotated_feature_align_backward_cpu_parrots)
|
||||||
#ifdef MMCV_WITH_CUDA
|
#ifdef MMCV_WITH_CUDA
|
||||||
.apply(rotated_feature_align_backward_cuda_parrots)
|
.apply(rotated_feature_align_backward_cuda_parrots)
|
||||||
#endif
|
#endif
|
||||||
|
|
Loading…
Reference in New Issue