lukasplevac/cuda-samples
1
0
Fork 0
mirror of https://github.com/NVIDIA/cuda-samples.git synced 2025-04-10 10:12:11 +01:00
Code Issues Packages Projects Releases Wiki Activity

Adding suport for Blackwell

Browse Source
This commit is contained in:
shawnz 2025-02-05 11:17:06 +08:00
parent e8a041d783
commit 78f83ca02d
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}, {0x87, 128},
{0x89, 128}, {0x89, 128},
{0x90, 128}, {0x90, 128},
{0xa0, 128},
{0xa1, 128},
{0xc0, 128},
{-1, -1}}; {-1, -1}};
int index = 0; int index = 0;
@ -717,6 +720,9 @@ inline const char* _ConvertSMVer2ArchName(int major, int minor) {
{0x87, "Ampere"}, {0x87, "Ampere"},
{0x89, "Ada"}, {0x89, "Ada"},
{0x90, "Hopper"}, {0x90, "Hopper"},
{0xa0, "Blackwell"},
{0xa1, "Blackwell"},
{0xc0, "Blackwell"},
{-1, "Graphics Device"}}; {-1, "Graphics Device"}};
int index = 0; int index = 0;

4
Common/helper_cuda_drvapi.h
View File

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