From 6ddbda1c3354e222de821f9e0c3528ac8994f481 Mon Sep 17 00:00:00 2001 From: cDc Date: Sun, 29 Dec 2024 12:27:01 +0200 Subject: [PATCH] common: add support for new GPU architectures in CUDA core conversion --- libs/Common/UtilCUDA.cpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/libs/Common/UtilCUDA.cpp b/libs/Common/UtilCUDA.cpp index 5b79679dc..ca0405105 100644 --- a/libs/Common/UtilCUDA.cpp +++ b/libs/Common/UtilCUDA.cpp @@ -51,6 +51,9 @@ int _convertSMVer2Cores(int major, int minor) {0x72, 64 }, // Volta Generation (SM 7.2) GV10B class {0x75, 64 }, // Turing Generation (SM 7.5) TU1xx class {0x80, 64 }, // Ampere Generation (SM 8.0) GA100 class + {0x86, 128}, // Ampere Generation (SM 8.6) GA102-GA107 class + {0x89, 128}, // Ada Generation (SM 8.9) AD106 class + {0x90, 128}, // Hopper Generation (SM 9.0) GH100 class {-1, -1} };