Skip to content

Make changes to non-paged attention path#3

Open
sudhakarsingh27 wants to merge 4 commits into
cyanguwa:paged_attentionfrom
sudhakarsingh27:pr-1355-updates
Open

Make changes to non-paged attention path#3
sudhakarsingh27 wants to merge 4 commits into
cyanguwa:paged_attentionfrom
sudhakarsingh27:pr-1355-updates

Conversation

@sudhakarsingh27

Copy link
Copy Markdown
Collaborator

Description

Non-paged attention path needed to be changed to make the gemma tutorial work

Signed-off-by: Sudhakar Singh <sudhakars@nvidia.com>
Comment thread transformer_engine/pytorch/attention.py Outdated
Signed-off-by: Sudhakar Singh <sudhakars@nvidia.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