Skip to content

Turn off default async amax reduction - #148

Merged
ksivaman merged 1 commit into
NVIDIA:mainfrom
ksivaman:no_default_async_amax_reduce
Apr 14, 2023
Merged

Turn off default async amax reduction#148
ksivaman merged 1 commit into
NVIDIA:mainfrom
ksivaman:no_default_async_amax_reduce

Conversation

@ksivaman

Copy link
Copy Markdown
Member

No description provided.

Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

@timmoon10timmoon10 left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

:shipit:

@ksivaman

Copy link
Copy Markdown
MemberAuthor

/te-ci

1 similar comment
@ksivaman

Copy link
Copy Markdown
MemberAuthor

/te-ci

@ksivaman
ksivaman merged commit 9bc9e68 into NVIDIA:mainApr 14, 2023
erhoo82 pushed a commit to ksivaman/TransformerEngine-1 that referenced this pull request Apr 16, 2023
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
ksivaman added a commit that referenced this pull request Apr 19, 2023
* Port initial changes
Co-authored-by: Sangkug Lym <slym@nvidia.com>
Co-authored-by: Vasudevan Rengasamy <vrengasamy@nvidia.com>
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* readd FA include for PyTorch
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* Re-enable sm_70 + cleanup
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* LICENSE, cleanup header
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* 5k -> 173 errors
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* license and fixes in userbuffers-host
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* next round fixes
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* final cpp cleanup
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* pylinting
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* fix from linting
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* Turn off default async amax reduction (#148)
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* remove unused code path
Signed-off-by: Sangkug Lym <slym@nvidia.com>
* cleanup Macros
Signed-off-by: Sangkug Lym <slym@nvidia.com>
* fix conflict resolution bug
Signed-off-by: Sangkug Lym <slym@nvidia.com>
* Fix gencode flags in setup (#145)
* Fix gencode flags based on cuda version
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* review suggestions
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* revert append_nvcc_threads change
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
---------
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* Change overlap config dict error message
Signed-off-by: Sangkug Lym <slym@nvidia.com>
* simplify ub initialization
Signed-off-by: Sangkug Lym <slym@nvidia.com>
* lint
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* fix sanity imports
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* cpplint
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* fix TensorFlow build
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* fix TE macros in public header
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* fix lint
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* More fixes
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* compiles with and w/o MPI
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* fixes for python side annotations for conditional compile
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* link gdrAPI only when MPI found
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* fix comments for dummy var
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* Fix linking
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* Review comments
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* load MPI before TE
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* Add Py side argument checks
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* remove unused code and catch silent failures
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* Fix cpp tests
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* fix find_lib path for tests
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
---------
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Signed-off-by: Sangkug Lym <slym@nvidia.com>
Co-authored-by: Sangkug Lym <slym@nvidia.com>
Co-authored-by: Vasudevan Rengasamy <vrengasamy@nvidia.com>
ptrendx pushed a commit that referenced this pull request Apr 25, 2023
* Port initial changes
Co-authored-by: Sangkug Lym <slym@nvidia.com>
Co-authored-by: Vasudevan Rengasamy <vrengasamy@nvidia.com>
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* readd FA include for PyTorch
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* Re-enable sm_70 + cleanup
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* LICENSE, cleanup header
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* 5k -> 173 errors
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* license and fixes in userbuffers-host
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* next round fixes
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* final cpp cleanup
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* pylinting
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* fix from linting
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* Turn off default async amax reduction (#148)
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* remove unused code path
Signed-off-by: Sangkug Lym <slym@nvidia.com>
* cleanup Macros
Signed-off-by: Sangkug Lym <slym@nvidia.com>
* fix conflict resolution bug
Signed-off-by: Sangkug Lym <slym@nvidia.com>
* Fix gencode flags in setup (#145)
* Fix gencode flags based on cuda version
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* review suggestions
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* revert append_nvcc_threads change
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
---------
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* Change overlap config dict error message
Signed-off-by: Sangkug Lym <slym@nvidia.com>
* simplify ub initialization
Signed-off-by: Sangkug Lym <slym@nvidia.com>
* lint
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* fix sanity imports
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* cpplint
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* fix TensorFlow build
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* fix TE macros in public header
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* fix lint
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* More fixes
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* compiles with and w/o MPI
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* fixes for python side annotations for conditional compile
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* link gdrAPI only when MPI found
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* fix comments for dummy var
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* Fix linking
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* Review comments
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* load MPI before TE
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* Add Py side argument checks
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* remove unused code and catch silent failures
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* Fix cpp tests
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
* fix find_lib path for tests
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
---------
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Signed-off-by: Sangkug Lym <slym@nvidia.com>
Co-authored-by: Sangkug Lym <slym@nvidia.com>
Co-authored-by: Vasudevan Rengasamy <vrengasamy@nvidia.com>
@ksivaman
ksivaman deleted the no_default_async_amax_reduce branch July 19, 2023 01:39
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants

@ksivaman@timmoon10