Dockerfile - Support cuda12.8 for Blackwell arch #682
Conversation
dpower4
force-pushed
the
feat/GB_12_8_build
branch
from
d963ad8 to
6a51193
Compare
Codecov ReportAll modified and coverable lines are covered by tests ✅
Additional details and impacted files@@ Coverage Diff @@
## main #682 +/- ##
=======================================
Coverage 85.67% 85.67%
=======================================
Files 99 99
Lines 7211 7211
=======================================
Hits 6178 6178
Misses 1033 1033
Flags with carried forward coverage won't be shown. Click here to find out more. ☔ View full report in Codecov by Sentry. 🚀 New features to boost your workflow:
|
| $(eval ARCHS := 75 80 86 89 90 100) | ||
| if [ -d msccl ]; then rm -rf msccl; fi; \ | ||
| git clone --single-branch --branch main https://github.com/Azure/msccl.git \ | ||
| && git -C msccl checkout 87048bd && git -C msccl submodule update --recursive --init | ||
| else ifeq ($(shell echo $(CUDA_VER)">=11.8" | bc -l), 1) | ||
| $(eval ARCHS := 70 75 80 86 89 90) |
There was a problem hiding this comment.
may only need 80, 90, and 100 archs for build, didn't test on the rest
There was a problem hiding this comment.
Building for the ARCH's for which cuda version supports. Will be helpful if someone wants to use this container on a 75 arch.
On the side: condensing the list will definitely help reduce the overall build time.
There was a problem hiding this comment.
Actually I mean we didn't test msccl on V100 (75 arch), A10 (86 arch) etc. so it may not work
abuccts
changed the title
Description
Updated docker for 12.8
Use cutlass latest relase 3.8 with ARCH 100(blackwell) support
add latest nccl-test release with ARCH 100(blackwell)
Updated msccl to support build for sm_100
No breaking changes, so backward compatible tested with 12.4(for now)