Skip to content

Ring Attention First Pass Implementation#407

Open
pulitha13 wants to merge 1 commit into
foundation-model-stack:mainfrom
coms-6998-context-parallelism-project:fms-pr-squashed
Open

Ring Attention First Pass Implementation#407
pulitha13 wants to merge 1 commit into
foundation-model-stack:mainfrom
coms-6998-context-parallelism-project:fms-pr-squashed

Conversation

@pulitha13
Copy link
Copy Markdown

This is a first pass in using context parallelism for FMS based on the ring attention paper here. We are treating this pull request as a work in progress. We would be happy to make any changes or fixes as IBM sees fit

Known issues

  • We are not perfectly generating 1:1 output logits when compared to the default configuration of llama 7b. We aim to figure out why our results are deviating yet producing "close enough" output.

Future improvements:

  • Online softmaxing during ring attention

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