Skip to content
This repository has been archived by the owner on Dec 1, 2021. It is now read-only.

Optimize batchnorm #469

Merged
merged 2 commits into from
Oct 1, 2019
Merged

Optimize batchnorm #469

merged 2 commits into from
Oct 1, 2019

Conversation

primenumber
Copy link
Contributor

Batch Normalization about 50% faster!

Motivation and Context

This PR optimize batch normalization.

Description

  • Allocate buffer in initialization
  • Use AVX if enabled
  • Use std::unique_ptr

How has this been tested?

Screenshots (if appropriate):

Types of changes

  • Bug fix (non-breaking change which fixes an issue)
  • New feature / Optimization (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)

Checklist:

  • My change requires a change to the documentation.
  • I have updated the documentation accordingly.

@primenumber primenumber self-assigned this Sep 25, 2019
@blueoil-butler blueoil-butler bot added the CI: auto-run Run CI automatically label Sep 25, 2019
tkng
tkng approved these changes Oct 1, 2019
View reviewed changes
Copy link
Contributor

@tkng tkng left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@tkng tkng merged commit 2ee4a41 into blue-oil:master Oct 1, 2019
@primenumber primenumber deleted the optimize/batchnorm branch October 1, 2019 06:37
@iizukak iizukak added this to the v0.12.0 milestone Oct 3, 2019
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Reviewers

@tkng tkng tkng approved these changes

Assignees

@primenumber primenumber

Labels
CI: auto-run Run CI automatically
Projects
None yet
Milestone
v0.12.0
Development

Successfully merging this pull request may close these issues.
3 participants
@primenumber @tkng @iizukak