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

Piz/hybridmesh #66

Open
wants to merge 1 commit into
base: flash_attention
Choose a base branch
from
Open

Piz/hybridmesh #66

wants to merge 1 commit into from

Conversation

zpcore
Copy link
Collaborator

@zpcore zpcore commented Oct 24, 2024

What does this PR do?

Use hybrid mesh based on this PR. This is needed for TPU multislice workload.

@zpcore zpcore changed the base branch from main to flash_attention October 24, 2024 18:05
@zpcore zpcore force-pushed the piz/hybridmesh branch 2 times, most recently from 4b8e8bf to c5ba21b Compare October 28, 2024 23:51
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