laptop gpu device capabilites

6db73fab1cc5cd885b585fa64d766c7304a69ff8 · 2024-08-21 21:33:29 +0100 · Alex Cheema

Files touched

M exo/topology/device_capabilities.py

Diff

commit 6db73fab1cc5cd885b585fa64d766c7304a69ff8
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Wed Aug 21 21:33:29 2024 +0100

    laptop gpu device capabilites
---
 exo/topology/device_capabilities.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/exo/topology/device_capabilities.py b/exo/topology/device_capabilities.py
index 03dccfc8..ccae1696 100644
--- a/exo/topology/device_capabilities.py
+++ b/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:

← 647ffb94 increase cli generation timeout · back to Exo · keep 4 in RotatingKVCache 5101f033 →