Skip to content

Commit af3d275

Browse files
committed
Add return type annotations to version.pyx; fix minor arithmetic
Add -> tuple[int, int, int] annotations to binding_version and driver_version. Align driver_version arithmetic with _system.pyx. Made-with: Cursor
1 parent 024ede1 commit af3d275

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

cuda_core/cuda/core/_utils/version.pyx

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ from cuda.core._utils.cuda_utils import driver, handle_return
99

1010

1111
@functools.cache
12-
def binding_version():
12+
def binding_version() -> tuple[int, int, int]:
1313
"""Return the cuda-bindings version as a (major, minor, patch) triple."""
1414
try:
1515
parts = importlib.metadata.version("cuda-bindings").split(".")[:3]
@@ -19,10 +19,10 @@ def binding_version():
1919

2020

2121
@functools.cache
22-
def driver_version():
22+
def driver_version() -> tuple[int, int, int]:
2323
"""Return the CUDA driver version as a (major, minor, patch) triple."""
2424
cdef int ver = handle_return(driver.cuDriverGetVersion())
25-
return (ver // 1000, (ver % 1000) // 10, ver % 10)
25+
return (ver // 1000, (ver // 10) % 100, ver % 10)
2626

2727

2828
cdef tuple _cached_binding_version = None

0 commit comments

Comments
 (0)