DeepSpeed
f4caed6d - Allow NVIDIA Blackwell (#6991)

Commit
1 year ago
Allow NVIDIA Blackwell (#6991) NVIDIA Blackwell GPU generation has number 10. The SM code and architecture should be `100`, but the current code generates `1.`, because it expects a 2 characters string. This change modifies the logic to consider it as a string that contains a `.`, hence splits the string and uses the array of strings. Signed-off-by: Fabien Dupont <fdupont@redhat.com>
Author
Parents
Loading