diff --git a/Common/helper_cuda.h b/Common/helper_cuda.h index 3c70fdc2..9440235e 100644 --- a/Common/helper_cuda.h +++ b/Common/helper_cuda.h @@ -673,7 +673,10 @@ inline int _ConvertSMVer2Cores(int major, int minor) { {0x90, 128}, {0xa0, 128}, {0xa1, 128}, + {0xa3, 128}, + {0xb0, 128}, {0xc0, 128}, + {0xc1, 128}, {-1, -1}}; int index = 0; @@ -725,7 +728,10 @@ inline const char* _ConvertSMVer2ArchName(int major, int minor) { {0x90, "Hopper"}, {0xa0, "Blackwell"}, {0xa1, "Blackwell"}, + {0xa3, "Blackwell"}, + {0xb0, "Blackwell"}, {0xc0, "Blackwell"}, + {0xc1, "Blackwell"}, {-1, "Graphics Device"}}; int index = 0; diff --git a/Common/helper_cuda_drvapi.h b/Common/helper_cuda_drvapi.h index 3303e6f5..2ded503e 100644 --- a/Common/helper_cuda_drvapi.h +++ b/Common/helper_cuda_drvapi.h @@ -118,7 +118,10 @@ inline int _ConvertSMVer2CoresDRV(int major, int minor) { {0x90, 128}, {0xa0, 128}, {0xa1, 128}, + {0xa3, 128}, + {0xb0, 128}, {0xc0, 128}, + {0xc1, 128}, {-1, -1}}; int index = 0;