LAPTOP GPU and Laptop GPU prefixes

e8267e73875ef324d42c6936182a893c3de44ac8 · 2024-08-10 12:59:27 +0100 · Alex Cheema

Files touched

M exo/topology/device_capabilities.py

Diff

commit e8267e73875ef324d42c6936182a893c3de44ac8
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Sat Aug 10 12:59:27 2024 +0100

    LAPTOP GPU and Laptop GPU prefixes
---
 exo/topology/device_capabilities.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/exo/topology/device_capabilities.py b/exo/topology/device_capabilities.py
index 02e26d72..110b5fbe 100644
--- a/exo/topology/device_capabilities.py
+++ b/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:

← 31641d10 tinygrad select model size · back to Exo · spacing 8e6414b2 →