AMD: Fix experimental/supported status output in GPU_Stats

o2team · Aug 5, 2019 · 275be7a · 275be7a
1 parent 45424d4
commit 275be7a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/plaidml_tools.py b/lib/plaidml_tools.py
@@ -73,8 +73,8 @@ def names(self):
         return ["{} - {} ({})".format(
             device.get("vendor", "unknown"),
             device.get("name", "unknown"),
-            "supported" if device in self.supported_devices else "experimental")
-                for device in self.device_details]
+            "supported" if idx in self.supported_indices else "experimental")
+                for idx, device in enumerate(self.device_details)]
 
     @property
     def supported_indices(self):