typo

e93a319ca4589d0a689eb6e8db195920bbb98475 · 2024-07-16 18:34:13 -0700 · Alex Cheema

Files touched

M exo/topology/device_capabilities.py

Diff

commit e93a319ca4589d0a689eb6e8db195920bbb98475
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Tue Jul 16 18:34:13 2024 -0700

    typo
---
 exo/topology/device_capabilities.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/exo/topology/device_capabilities.py b/exo/topology/device_capabilities.py
index eb8d206d..fa74579f 100644
--- a/exo/topology/device_capabilities.py
+++ b/exo/topology/device_capabilities.py
@@ -43,7 +43,7 @@ def linux_device_capabilities() -> DeviceCapabilities:
     from tinygrad import Device
     
     print(f"tinygrad {Device.DEFAULT=}")
-    if Device.DEFAULT == "CUDA" or Device.DEFAULT == "NV" or Device.DEFAULT="GPU":
+    if Device.DEFAULT == "CUDA" or Device.DEFAULT == "NV" or Device.DEFAULT=="GPU":
         import pynvml, pynvml_utils
         pynvml.nvmlInit()
         handle = pynvml.nvmlDeviceGetHandleByIndex(0)

← ce46f000 linux device capabilities · back to Exo · fix linux/amd gpu memory, convert to MB c819f675 →