Skip to content

Conversation

@jeffra
Copy link
Collaborator

@jeffra jeffra commented May 24, 2021

  • Add requirement for dev on importlib-metadata > 4, we were seeing pytest issues with importlib-metadata 2.0
  • Delay import of static class variables, this was causing our unit test framework to crash since importing megatron seems to initialize cuda which causes us python/pytorch multi-process issues :(
  • fix incorrect import path for ParallelTransformerLayer

@jeffra jeffra merged commit 96eb5b1 into master May 24, 2021
@jeffra jeffra deleted the jeffra/delay-imports branch May 24, 2021 23:43
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.

3 participants