Browse Source

LAPTOP GPU and Laptop GPU prefixes

Alex Cheema 1 năm trước cách đây
mục cha
commit
e8267e7387
1 tập tin đã thay đổi với 3 bổ sung0 xóa
  1. 3 0
      exo/topology/device_capabilities.py

+ 3 - 0
exo/topology/device_capabilities.py

@@ -113,6 +113,9 @@ CHIP_FLOPS = {
   # ... add more devices if needed ...
   ### Qualcomm embedded chips: TODO
 }
+CHIP_FLOPS.update({f"LAPTOP GPU {key}": value for key, value in CHIP_FLOPS.items()})
+CHIP_FLOPS.update({f"Laptop GPU {key}": value for key, value in CHIP_FLOPS.items()})
+
 
 
 def device_capabilities() -> DeviceCapabilities: