Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fixing error condition check when device microbatch size times seq parallelism dim is not 1 due to floating point precision #3200

Conversation

ShashankMosaicML
Copy link
Contributor

@ShashankMosaicML ShashankMosaicML commented Apr 19, 2024

What does this PR do?

Due to floating point precision, if the device microbatch size times the sequence parallel dimension is not exactly 1, then this error condition is triggered. This PR allows for some error due to floating point precision.

ShashankMosaicML and others added 3 commits April 18, 2024 23:01
…_parallelism' of github.com:mosaicml/composer into shashank/fix_error_condition_device_microbatch_size_seq_parallelism
@ShashankMosaicML ShashankMosaicML marked this pull request as ready for review April 19, 2024 15:15
@ShashankMosaicML ShashankMosaicML merged commit 7b25d90 into dev Apr 19, 2024
14 checks passed
@ShashankMosaicML ShashankMosaicML deleted the shashank/fix_error_condition_device_microbatch_size_seq_parallelism branch April 19, 2024 16:59
DhruvDh pushed a commit to DhruvDh/composer that referenced this pull request Apr 21, 2024
…rallelism dim is not 1 due to floating point precision (mosaicml#3200)

* ..

* ..

* lint
j316chuck pushed a commit that referenced this pull request May 16, 2024
…rallelism dim is not 1 due to floating point precision (#3200)

* ..

* ..

* lint
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.

2 participants