bigscience-workshop / bigscience-workshop/Megatron-DeepSpeed

Parallelize Meg CUDA Kernel build system

Open
#174 0 comments 0 reactions 0 assignees View on GitHub
Good Difficult Issue Good First Issue
Dominant language
Python
Stars
1.4k
Forks
226
PR merge metrics
No merged PRs in 30d

Description

It takes forever to build the Meg cuda kernels as it does it sequentially and doesn't take advantage of multiple cores. It takes some 5 minutes to build. And every time one changes the number of gpus it rebuilds itself, which is both very non-productive and it also makes the CI really slow.

Need to rewrite the build to parallelize it.

Sidenotes: apex and deepspeed have this too, but deepspeed supports `make -j`

And ideally the solution needs to come from pytorch, perhaps if we solve it generically we could upstream the solution to pytorch core.

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.