Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Add omp parallel optimization for _contrib_BilinearReisze2D #15584

Merged
merged 4 commits into from
Jul 21, 2019

Conversation

wuxun-zhang
Copy link
Contributor

Description

This PR aims to improve the performance of _contrib_BilinearReisze2D via omp parallel optimization.
This will be a great help to deploy Fully Convolutional Network model on CPU. @pengzhao-intel @ZhennanQin @TaoLv @zhanghang1989

The below table shows the speedup between w/ OMP and w/o OMP.

Before resize After resize w/o OMP  (ms) w/ OMP 28 cores (ms) Speedup
shape: (1, 32, 32, 32) shape: (1, 32, 480, 480) 25.643921 5.877519 4.36
shape: (2, 64, 32, 32) shape: (2, 64, 480, 480) 134.311175 32.115507 4.18
shape: (1, 128, 64, 64) shape: (1, 128, 480, 480) 126.516438 34.05602 3.71
shape: (32, 32, 32, 32) shape: (32, 32, 480, 480) 2262.586236 341.702771 6.62
shape: (32, 32, 64, 64) shape: (32, 32, 480, 480) 2201.238561 339.254451 6.49

Checklist

Essentials

  • Changes are complete (i.e. I finished coding on this PR)
  • All changes have test coverage:
  • To the my best knowledge, examples are either not affected by this change, or have been fixed to be compatible with this change

@pengzhao-intel
Copy link
Contributor

cc @zhreshold

@zhanghang1989
Copy link
Contributor

Great job.

@karan6181
Copy link
Contributor

@mxnet-label-bot add [Operator, pr-awaiting-review]

@marcoabreu marcoabreu added Operator pr-awaiting-review PR is waiting for code review labels Jul 19, 2019
@marcoabreu marcoabreu merged commit 9a1a102 into apache:master Jul 21, 2019
@wuxun-zhang wuxun-zhang deleted the improve_bilinear_resize_op branch July 26, 2019 01:38
anirudhacharya pushed a commit to anirudhacharya/mxnet that referenced this pull request Aug 20, 2019
…5584)

* Add omp parallel optimization for bilinear_resize op

* retrigger CI

* retrigger CI

* trigger CI
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
Operator pr-awaiting-review PR is waiting for code review
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants