add fp16 support for forwarding (b04a0157) · Commits · Joel Joseph / food-round2 · GitLab

Skip to content

GitLab

Explore

Sign in

Primary navigation

Project

F

food-round2
- Activity
- Members
- Labels
- Issues
- Issue boards
- Milestones
- Wiki
- Releases
- Model registry
- Environments
- Incidents

Snippets Groups Projects

b04a0157

Commit b04a0157 authored 6 years ago by Kai Chen

Downloads
- Patches
- Plain Diff

add fp16 support for forwarding

parent 7f9d2eb5

No related branches found

No related tags found

No related merge requests found

Changes 3

Hide whitespace changes

Inline Side-by-side

Showing

mmdet/ops/roi_align/src/roi_align_kernel.cu 40 additions, 52 deletions

mmdet/ops/roi_align/src/roi_align_kernel.cu
mmdet/ops/roi_pool/gradcheck.py 1 addition, 1 deletion

mmdet/ops/roi_pool/gradcheck.py
mmdet/ops/roi_pool/src/roi_pool_kernel.cu 20 additions, 45 deletions

mmdet/ops/roi_pool/src/roi_pool_kernel.cu

with 61 additions and 98 deletions

Loading

0% Loading or .

You are about to add 0 people to the discussion. Proceed with caution.

Finish editing this message first!

Please register or sign in to comment