tests: Report Blackwell architecture

jp7677 · Jan 9, 2025 · c9651dd · c9651dd
1 parent 40b5da3
commit c9651dd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/nvapi_system.cpp b/tests/nvapi_system.cpp
@@ -30,6 +30,8 @@ static std::string ToGpuType(NV_GPU_TYPE type) {
 
 static std::string ToGpuArchitecture(NV_GPU_ARCHITECTURE_ID id) {
     switch (id) {
+        case NV_GPU_ARCHITECTURE_GB200:
+            return "Blackwell";
         case NV_GPU_ARCHITECTURE_AD100:
             return "Ada";
         case NV_GPU_ARCHITECTURE_GA100: