Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
MMCV
Commits
78fdf178
Unverified
Commit
78fdf178
authored
Jan 10, 2023
by
CokeDong
Committed by
GitHub
Jan 10, 2023
Browse files
Update parrots image (#2538)
* Update image for 2.x * Update install cmd for 2.x
parent
16155078
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
18 additions
and
7 deletions
+18
-7
.github/workflows/build_pat.yml
.github/workflows/build_pat.yml
+14
-5
mmcv/ops/csrc/parrots/border_align_parrots.cpp
mmcv/ops/csrc/parrots/border_align_parrots.cpp
+2
-0
mmcv/ops/csrc/parrots/rotated_feature_align_parrots.cpp
mmcv/ops/csrc/parrots/rotated_feature_align_parrots.cpp
+2
-2
No files found.
.github/workflows/build_pat.yml
View file @
78fdf178
name
:
build_pat
on
:
push
on
:
push
:
paths-ignore
:
-
'
README.md'
-
'
README_zh-CN.md'
-
'
docs/**'
-
'
examples/**'
-
'
.dev_scripts/**'
concurrency
:
group
:
${{ github.workflow }}-${{ github.ref }}
...
...
@@ -13,14 +20,16 @@ jobs:
build_parrots
:
runs-on
:
ubuntu-18.04
container
:
image
:
ghcr.io/
zhouzaida/parrots-mmcv:1.3.4
image
:
ghcr.io/
cokedong/parrots:pat0.21.0a0_cuda11
credentials
:
username
:
zhouzaida
password
:
${{ secrets.
CR_PAT
}}
username
:
cokedong
password
:
${{ secrets.
PARROTS_CI_TOKEN
}}
steps
:
-
uses
:
actions/checkout@v2
-
name
:
Install unittest dependencies
run
:
pip install -r requirements/test.txt
-
name
:
Build and install
run
:
rm -rf .eggs && MMCV_WITH_OPS=1 pip install -e .
run
:
|
source /usr/local/env/pat_latest
pip install -e . -v
mmcv/ops/csrc/parrots/border_align_parrots.cpp
View file @
78fdf178
...
...
@@ -7,6 +7,7 @@
using
namespace
parrots
;
#ifdef MMCV_WITH_CUDA
void
border_align_forward_cuda_parrots
(
CudaContext
&
ctx
,
const
SSElement
&
attr
,
const
OperatorBase
::
in_list_t
&
ins
,
OperatorBase
::
out_list_t
&
outs
)
{
...
...
@@ -49,3 +50,4 @@ PARROTS_EXTENSION_REGISTER(border_align_backward)
.
output
(
1
)
.
apply
(
border_align_backward_cuda_parrots
)
.
done
();
#endif
mmcv/ops/csrc/parrots/rotated_feature_align_parrots.cpp
View file @
78fdf178
...
...
@@ -40,6 +40,7 @@ void rotated_feature_align_backward_cuda_parrots(
rotated_feature_align_backward
(
grad_output
,
best_bboxes
,
grad_input
,
spatial_scale
,
points
);
}
#endif
void
rotated_feature_align_forward_cpu_parrots
(
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
,
points
);
}
#endif
void
rotated_feature_align_backward_cpu_parrots
(
HostContext
&
ctx
,
const
SSElement
&
attr
,
const
OperatorBase
::
in_list_t
&
ins
,
...
...
@@ -92,7 +92,7 @@ PARROTS_EXTENSION_REGISTER(rotated_feature_align_backward)
.
attr
(
"points"
)
.
input
(
2
)
.
output
(
1
)
.
apply
(
rotated_feature_align_
for
ward_cpu_parrots
)
.
apply
(
rotated_feature_align_
back
ward_cpu_parrots
)
#ifdef MMCV_WITH_CUDA
.
apply
(
rotated_feature_align_backward_cuda_parrots
)
#endif
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment