From 0344d84f13c509741ba5b5d12deb35e49029ceda Mon Sep 17 00:00:00 2001
From: Qianqian Fang <fangqq@gmail.com>
Date: Mon, 4 Mar 2024 00:04:10 -0500
Subject: [PATCH] [bug] fix cuda core count for Ada and Blackwell

---
 src/mcx_core.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/mcx_core.cu b/src/mcx_core.cu
index c970ad02..8bed2a3a 100644
--- a/src/mcx_core.cu
+++ b/src/mcx_core.cu
@@ -2428,7 +2428,7 @@ int mcx_corecount(int v1, int v2) {
         return 48;
     } else if (v < 50) {
         return 192;
-    } else if (v < 60 || v == 61) {
+    } else if (v < 60 || v == 61 || v >= 89) {
         return 128;
     } else {
         return 64;