Skip to content

Added additional check to run with distributed enabled and world_size=1 (#96)#116

Merged
1 commit merged into
HabanaAI:habana-mainfrom
kalyanjk:v1.15-synapse_mark_step
Mar 18, 2024
Merged

Added additional check to run with distributed enabled and world_size=1 (#96)#116
1 commit merged into
HabanaAI:habana-mainfrom
kalyanjk:v1.15-synapse_mark_step

Conversation

@kalyanjk
Copy link
Copy Markdown

Added additional check to run with distributed enabled and  world_size=1 (#96)

* Added additional check to run with distributed enabled and  world_size = 1
* Reduce the number of graph splits to avoid memory allocation error  for 1x LLAMA1_7b_ft

…=1 (HabanaAI#96)

* Added additionla check to run with distributed enabled and  world_size = 1

* Reduce the number of graph splits to avoid memory allocation error
for 1x LLAMA1_7b_ft

---------

Co-authored-by: Kalyan <kkumar@habana.ai>
@kalyanjk kalyanjk requested a review from a user March 18, 2024 12:41
@ghost ghost merged commit ea38ffb into HabanaAI:habana-main Mar 18, 2024
astachowiczhabana pushed a commit that referenced this pull request Apr 5, 2024
…=1 (#96) (#116)

* Added additionla check to run with distributed enabled and  world_size = 1

* Reduce the number of graph splits to avoid memory allocation error
for 1x LLAMA1_7b_ft

---------

Co-authored-by: Kalyan <kkumar@habana.ai>
astachowiczhabana pushed a commit that referenced this pull request Apr 5, 2024
…=1 (#96) (#116)

* Added additionla check to run with distributed enabled and  world_size = 1

* Reduce the number of graph splits to avoid memory allocation error
for 1x LLAMA1_7b_ft

---------

Co-authored-by: Kalyan <kkumar@habana.ai>
@astachowiczhabana
Copy link
Copy Markdown

huggingface#675

@kalyanjk
Copy link
Copy Markdown
Author

This PR solves the actual issue #126

This pull request was closed.
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