소스 검색

laptop gpu device capabilites

Alex Cheema 11 달 전
부모
커밋
6db73fab1c
1개의 변경된 파일2개의 추가작업 그리고 0개의 파일을 삭제
  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: