From c30e045b8fba87d7fb1e4fceb8ea6cd2facec8aa Mon Sep 17 00:00:00 2001 From: Anatoly Urban Date: Mon, 24 Apr 2023 20:02:25 +0300 Subject: [PATCH] Fix number of cores for CC86-89 --- src/cudart/gpucard.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/cudart/gpucard.cpp b/src/cudart/gpucard.cpp index d1dd8d50d..e6a0ca015 100644 --- a/src/cudart/gpucard.cpp +++ b/src/cudart/gpucard.cpp @@ -133,6 +133,8 @@ static void getDeviceAttribute(DeviceAttribute& a, int device = 0) found_cc = false; else if (a.cc >= 90) a.cores_per_multiprocessor = 128; + else if (a.cc >= 86) + a.cores_per_multiprocessor = 128; else if (a.cc >= 80) a.cores_per_multiprocessor = 64; else if (a.cc >= 61)