From 1600466914881e7c8ef3e7e8996cb0a77fd171ab Mon Sep 17 00:00:00 2001
From: RichardHaselgrove <r.haselgrove@btinternet.com>
Date: Fri, 25 Sep 2020 17:47:00 +0100
Subject: [PATCH] client/scheduler: update NVidia detection again

for new RTX 30x0 range released September 2020
---
 lib/coproc.cpp | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/lib/coproc.cpp b/lib/coproc.cpp
index 19121ec458..083288c2c0 100644
--- a/lib/coproc.cpp
+++ b/lib/coproc.cpp
@@ -565,10 +565,21 @@ void COPROC_NVIDIA::set_peak_flops() {
             }
             break;
         case 7:    // for both cc7.0 (Titan V, Tesla V100) and cc7.5 (RTX, Tesla T4)
-        default:
             flops_per_clock = 2;
             cores_per_proc = 64;
             break;
+        case 8:    // for cc8.0 (A100) and cc8.6 (GeForce RTX 30x0 - GA102 and above)
+        default:
+            flops_per_clock = 2;
+            switch (minor) {
+            case 0:    // special for A100 Tensor Core datacenter GPU
+                cores_per_proc = 64;
+                break;
+            default:
+                cores_per_proc = 128;
+                break;
+            }
+            break;
         }
 
     }