Skip to content

[Pallas] Make gmm functional#7117

Merged
alanwaketan merged 10 commits intomasterfrom
alanwaketan/gmm3
May 26, 2024
Merged

[Pallas] Make gmm functional#7117
alanwaketan merged 10 commits intomasterfrom
alanwaketan/gmm3

Conversation

@alanwaketan
Copy link
Copy Markdown
Collaborator

Summary:
This should be the last PR needed to make gmm functional. Now, everything should be executed in the xla devices and produces the correct output.

Test Plan:
python test/test_gmm.py

@alanwaketan alanwaketan self-assigned this May 25, 2024
@JackCaoG
Copy link
Copy Markdown
Collaborator

AssertionError: "min must be less than max." does not match "min must be less than or equal to max." lol seems like you need to update the test.

@alanwaketan
Copy link
Copy Markdown
Collaborator Author

Thanks, Jack.

@alanwaketan alanwaketan merged commit a9b4fad into master May 26, 2024
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