Coverage for cuda/core/_utils/version.pyx: 92.86%

1# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. (empty)

2# 

3# SPDX-License-Identifier: Apache-2.0 

4  

5import functools (empty)

6import importlib.metadata (empty)

7import re (empty)

8  

9from cuda.core._utils.cuda_utils import driver, handle_return (empty)

10  

11  

12def _parse_version_triple(version_str: str) -> tuple[int, int, int]: (empty)

13 """Parse a PEP 440 version string into a (major, minor, patch) triple. 

14  

15 Strips local-version identifiers and handles pre-release suffixes such as 

16 ``0b1`` or ``0rc1`` by extracting only the leading integer from each 

17 release segment. 

18 """ 

19 parts = version_str.partition("+")[0].split(".")[:3] (empty)

20 ints = ([int(m.group(1)) if (m := re.match(r"(\d+)", v)) else 0 for v in parts] + [0, 0, 0])[:3] (empty)

21 return (ints[0], ints[1], ints[2]) (empty)

22  

23  

24@functools.cache (empty)

25def binding_version() -> tuple[int, int, int]: 

26 """Return the cuda-bindings version as a (major, minor, patch) triple.""" 

27 try: (empty)

28 version_str = importlib.metadata.version("cuda-bindings") (empty)

29 except importlib.metadata.PackageNotFoundError: 

30 version_str = importlib.metadata.version("cuda-python") 

31 return _parse_version_triple(version_str) (empty)

32  

33  

34@functools.cache (empty)

35def driver_version() -> tuple[int, int, int]: 

36 """Return the CUDA driver version as a (major, minor, patch) triple.""" 

37 cdef int ver = handle_return(driver.cuDriverGetVersion()) (empty)

38 return (ver // 1000, (ver // 10) % 100, ver % 10) (empty)

39  

40  

41cdef tuple _cached_binding_version = None (empty)

42cdef tuple _cached_driver_version = None (empty)

43  

44  

45cdef tuple cy_binding_version(): (empty)

46 global _cached_binding_version 

47 if _cached_binding_version is None: 235 ctx2AbBbc CbDbEbFbGbHbIbd e f JbKbLbMbNbObPbg Qbh i j k l m n o p q r s t u v w x y z A B C D E F G RbSbTbUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9b!b#b$b%b'b(b)b*b+byb,b-b.b/b:b;b=b?b@bH [b]b^b_b`b{b|b}b~bacbcccdcecfcI gchcicjckclcmcncocpcqcrcsctcucvcwcxcyczcAcBcCcDcEcFcGcHcJ K L M N O P Q R S T U V W X Y IcJcZ 0 1 Kc2 3 4 5 6 7 8 9 ! # $ % ' Lc( ) McNcOcPc* Qc+ RcScTcUcVc, - . / : ; = ? WcXc@ YcZc0c1c2c3c4c[ ] ^ _ ` { | } ~ 5cabbbcbdbebfbgbhbibjbkblbmbnbobpbqbrbsbtbb ubvbwbxb6c

48 _cached_binding_version = binding_version() 2 ctx2ybb

49 return _cached_binding_version 235 ctx2AbBbc CbDbEbFbGbHbIbd e f JbKbLbMbNbObPbg Qbh i j k l m n o p q r s t u v w x y z A B C D E F G RbSbTbUbVbWbXbYbZb0b1b2b3b4b5b6b7b8b9b!b#b$b%b'b(b)b*b+byb,b-b.b/b:b;b=b?b@bH [b]b^b_b`b{b|b}b~bacbcccdcecfcI gchcicjckclcmcncocpcqcrcsctcucvcwcxcyczcAcBcCcDcEcFcGcHcJ K L M N O P Q R S T U V W X Y IcJcZ 0 1 Kc2 3 4 5 6 7 8 9 ! # $ % ' Lc( ) McNcOcPc* Qc+ RcScTcUcVc, - . / : ; = ? WcXc@ YcZc0c1c2c3c4c[ ] ^ _ ` { | } ~ 5cabbbcbdbebfbgbhbibjbkblbmbnbobpbqbrbsbtbb ubvbwbxb6c

50  

51  

52cdef tuple cy_driver_version(): (empty)

53 global _cached_driver_version 

54 if _cached_driver_version is None: 132 ctx2c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ 7c8c9c!c#c$c%c'c(c)c*c+c,c-c.c/czb:c;c=c] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbqbrbsbtbb ubvbwbxb

55 _cached_driver_version = driver_version() 2 ctx2zbb

56 return _cached_driver_version 132 ctx2c d e f g h i j k l m n o p q r s t u v w x y z A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 0 1 2 3 4 5 6 7 8 9 ! # $ % ' ( ) * + , - . / : ; = ? @ [ 7c8c9c!c#c$c%c'c(c)c*c+c,c-c.c/czb:c;c=c] ^ _ ` { | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbqbrbsbtbb ubvbwbxb