Browse Source

laptop gpu device capabilites

Alex Cheema 10 months ago
parent
commit
6db73fab1c
1 changed files with 2 additions and 0 deletions
  1. 2 0
      exo/topology/device_capabilities.py

+ 2 - 0
exo/topology/device_capabilities.py

@@ -121,6 +121,8 @@ CHIP_FLOPS = {
 }
 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()})
+CHIP_FLOPS.update({f"{key} LAPTOP GPU": value for key, value in CHIP_FLOPS.items()})
+CHIP_FLOPS.update({f"{key} Laptop GPU": value for key, value in CHIP_FLOPS.items()})
 
 
 def device_capabilities() -> DeviceCapabilities: