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

AMP support for Numpy ops#19036

Merged

szha merged 8 commits intoapache:masterfrom mk-61:pr-amp-numpy-ops

Oct 2, 2020

+471-292

Commits on Sep 24, 2020

AMP support for numpy ops
Vladimir Cherepanov
committed
Move misplaced docstring
Vladimir Cherepanov
committed
Fix numpy submodule handling in AMP
Vladimir Cherepanov
committed

Commits on Sep 27, 2020

Fix module selection when AMP-patching numpy ops
Vladimir Cherepanov
committed

Commits on Sep 29, 2020

Check if a model exists in AMP init
Vladimir Cherepanov
committed
Make AMP loss scale public
Vladimir Cherepanov
committed

Commits on Oct 1, 2020

Re-enable a test
Vladimir Cherepanov
committed
Re-disable a text, with the original reason
Vladimir Cherepanov
committed