Alex Cheema пре 1 година
родитељ
комит
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)