NASP/cuda-samples
52
0
Fork 0
mirror of https://github.com/NVIDIA/cuda-samples.git synced 2025-11-04 15:47:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #16 from XSShawnZeng/master

Browse Source 
Adding support for Blackwell
This commit is contained in:
Rob Armstrong 2025-02-04 21:46:10 -08:00 committed by GitHub
commit 9d03b030a6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Common/helper_cuda.h
View File

@ -668,6 +668,9 @@ inline int _ConvertSMVer2Cores(int major, int minor) {
{0x87, 128},
{0x89, 128},
{0x90, 128},
{0xa0, 128},
{0xa1, 128},
{0xc0, 128},
{-1, -1}};
int index = 0;
@ -717,6 +720,9 @@ inline const char* _ConvertSMVer2ArchName(int major, int minor) {
{0x87, "Ampere"},
{0x89, "Ada"},
{0x90, "Hopper"},
{0xa0, "Blackwell"},
{0xa1, "Blackwell"},
{0xc0, "Blackwell"},
{-1, "Graphics Device"}};
int index = 0;

4
Common/helper_cuda_drvapi.h
View File

@ -116,6 +116,9 @@ inline int _ConvertSMVer2CoresDRV(int major, int minor) {
{0x87, 128},
{0x89, 128},
{0x90, 128},
{0xa0, 128},
{0xa1, 128},
{0xc0, 128},
{-1, -1}};
int index = 0;
@ -405,4 +408,3 @@ bool inline findFatbinPath(const char *module_file, std::string &module_path, ch
// end of CUDA Helper Functions
#endif // COMMON_HELPER_CUDA_DRVAPI_H_