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

fix neuron imports to make unit tests run #27

Conversation

lipovsek-aws
Copy link
Collaborator

@lipovsek-aws lipovsek-aws commented Dec 10, 2024

  • remove unneded top level imports, especially neuron ones as they are breaking GPU tests

@lipovsek-aws lipovsek-aws changed the title fix neuron attention to make unit tests run fix neuron imports to make unit tests run Dec 10, 2024
@patrick-toulme
Copy link
Owner

Thanks!

@patrick-toulme patrick-toulme merged commit cac3b1b into mainline_upstream_boilerplate_intermediate-2 Dec 10, 2024
0 of 11 checks passed
apivovarov pushed a commit that referenced this pull request Dec 10, 2024
* fix neuron attention to make unit tests run

* remove top level iport from flash attention utils

* revert LNC change
apoorvtintin pushed a commit to apoorvtintin/axlearn that referenced this pull request Dec 22, 2024
* fix neuron attention to make unit tests run

* remove top level iport from flash attention utils

* revert LNC change
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