Skip to content

Conversation

@stas00
Copy link
Contributor

@stas00 stas00 commented Sep 16, 2021

This PR

  • completes the torch_assert_equal wrapper to work with older pytorch version
  • adds torch_assert_close wrapper

Fixes: #52 (comment)

@stas00 stas00 merged commit fc8820c into bigscience-workshop:main Sep 16, 2021
@stas00 stas00 deleted the torch_all_close branch September 16, 2021 21:39
@stas00 stas00 mentioned this pull request Sep 16, 2021
5 tasks
ofirpress pushed a commit to ofirpress/Megatron-DeepSpeed that referenced this pull request Sep 23, 2021
SaulLu added a commit to SaulLu/Megatron-DeepSpeed that referenced this pull request Sep 24, 2021
adammoody pushed a commit to adammoody/Megatron-DeepSpeed that referenced this pull request Jun 21, 2023
…igscience-workshop#106)

use int64_t instead of int32_t to avoid integer overflow.

Signed-off-by: yulu.jia <yulu.jia@bytedance.com>
Co-authored-by: yulu.jia <yulu.jia@bytedance.com>
Sign up for free to 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.

1 participant