github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-07 22:23:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix linter errors

Browse Source
This commit is contained in:
Irwin1138
2025-11-23 22:23:26 +02:00
parent 7aea5256b1
commit 00aa943e91
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tinygrad/runtime/support/cuda.py
View File

@@ -1,4 +1,4 @@
import ctypes.util, os, platform
import os, platform
from ctypes.util import find_library
CUDA_PATH: str | None
@@ -10,8 +10,8 @@ if platform.system() == "Windows":
def find_nv_dll(glob_pattern):
cuda_bin = os.path.join(os.environ.get("CUDA_PATH", ""), "bin")
matches = glob.glob(os.path.join(cuda_bin, glob_pattern))
return matches[0] if matches else None
return matches[0] if matches else None
CUDA_PATH = find_library('nvcuda')
NVRTC_PATH = find_nv_dll("nvrtc64_*.dll")
NVJITLINK_PATH = find_nv_dll("nvJitLink_*.dll")