From 286d3cf1534727d2c98d0d885fe7c9fdeabc2954 Mon Sep 17 00:00:00 2001
From: Krzysztof Bogacki <krzysztof.bogacki@leancode.pl>
Date: Mon, 3 Feb 2025 20:11:28 +0100
Subject: [PATCH] tests: Update GPU arch reporting code for post-Blackwell

---
 tests/nvapi_system.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/nvapi_system.cpp b/tests/nvapi_system.cpp
index 21f90c6e..01b6b748 100644
--- a/tests/nvapi_system.cpp
+++ b/tests/nvapi_system.cpp
@@ -55,8 +55,8 @@ static std::string ToGpuArchitecture(NV_GPU_ARCHITECTURE_ID id) {
         case NV_GPU_ARCHITECTURE_GF110:
             return "Fermi";
         default:
-            if (id > NV_GPU_ARCHITECTURE_AD100)
-                return "Post-Ada";
+            if (id > NV_GPU_ARCHITECTURE_GB200)
+                return "Post-Blackwell";
             else
                 return "Pre-Fermi";
     }