Alex Cheema 10 ماه پیش
والد
کامیت
e93a319ca4
1فایلهای تغییر یافته به همراه1 افزوده شده و 1 حذف شده
  1. 1 1
      exo/topology/device_capabilities.py

+ 1 - 1
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)