浏览代码

LAPTOP GPU and Laptop GPU prefixes

Alex Cheema 1 年之前
父节点
当前提交
e8267e7387
共有 1 个文件被更改,包括 3 次插入0 次删除
  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: