tests: Update GPU arch reporting code for post-Blackwell

jp7677 · Feb 3, 2025 · 286d3cf · 286d3cf
1 parent 99f410e
commit 286d3cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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";
     }