Unverified Commit 64e739e0 authored by CokeDong's avatar CokeDong Committed by GitHub
Browse files

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 name
parent 859d84c5
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: ...@@ -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 @@ ...@@ -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) ...@@ -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( ...@@ -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( ...@@ -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) ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment