Skip to content

Commit

Permalink
feat: add nccl stream fetch api and add dependency version limit (#48)
Browse files Browse the repository at this point in the history 
1. add nccl stream fetch api in pytorch patches
2. add dependency version limit about numpy and pytest in torch_patch
and vescale requirements
  • Loading branch information
@vocaltract
vocaltract authored Aug 25, 2024
1 parent e439aa9 commit b4b1686
Show file tree
Hide file tree
Showing 2 changed files with 340 additions and 25 deletions.
Loading

0 comments on commit b4b1686

Please sign in to comment.