Coverage for cuda / bindings / _internal / _fast_enum.py: 100.00%

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

2# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE 

3 

4 

5# This code was automatically generated across versions from 12.9.1 to 13.1.1, generator version 0.3.1.dev1322+g646ce84ec. Do not modify it directly. 

6 

7 

8""" 

9This is a replacement for the stdlib enum.IntEnum. 

10 

11Notably, it has much better import time performance, since it doesn't generate 

12and evaluate Python code at startup time. 

13 

14It supports the most important subset of the IntEnum API. See `test_enum` in 

15`cuda_bindings/tests/test_basics.py` for details. 

16""" 

17 

18from typing import Any, Iterator (empty)

19 

20 

21class FastEnumMetaclass(type): (empty)

22 def __init__(cls, name, bases, namespace): (empty)

23 super().__init__(name, bases, namespace) 2 ctx1ab

24 

25 cls.__singletons__ = {} 2 ctx1ab

26 cls.__members__ = {} 2 ctx1ab

27 for name, value in cls.__dict__.items(): 2 ctx1ab

28 if name.startswith("__") and name.endswith("__"): 2 ctx1ab

29 continue 2 ctx1ab

30 

31 if isinstance(value, tuple): 2 ctx1ab

32 value, doc = value 2 ctx1ab

33 elif isinstance(value, int): (empty)

34 doc = None (empty)

35 else: 

36 continue (empty)

37 

38 singleton = int.__new__(cls, value) 2 ctx1ab

39 singleton.__doc__ = doc 2 ctx1ab

40 singleton._name = name 2 ctx1ab

41 cls.__singletons__[value] = singleton 2 ctx1ab

42 cls.__members__[name] = singleton 2 ctx1ab

43 

44 for name, member in cls.__members__.items(): 2 ctx1ab

45 setattr(cls, name, member) 2 ctx1ab

46 

47 def __repr__(cls) -> str: (empty)

48 return f"<enum '{cls.__name__}'>" 1 ctx1c

49 

50 def __len__(cls) -> int: (empty)

51 return len(cls.__members__) 4 ctx2d c LbMb

52 

53 def __iter__(cls) -> Iterator["FastEnum"]: (empty)

54 return iter(cls.__members__.values()) 10 ctx2a k d c l LbMbZg0g1g

55 

56 def __contains__(cls, item: Any) -> bool: (empty)

57 return item in cls.__singletons__ 1 ctx1c

58 

59 

60class FastEnum(int, metaclass=FastEnumMetaclass): (empty)

61 def __new__(cls, value: int) -> "FastEnum": (empty)

62 singleton: FastEnum = cls.__singletons__.get(value) 582 ctx2a NbObPbQbRbSbTbUbVbWbXbYbZb0b1b2bO P Q R S T U V W X Y 3b4b5b6b7b8b9b!b#b$b%b'b(b)b*b+b,b-b.b/b:b;b=b?b@b[b]b^b_b`b{b|b}b~bacZ 0 1 2 3 4 bcccdcecfcgc5 hc6 ic7 8 jckclc9 mc! ncocpcqcrcsctcucvcwc# xcyc$ zcAcBcCc% ' Dc( ) * + Ec, - . Fck Gc/ m HcIcJc: KcLcd c McNcOcPcl QcRcScTcUcVcWc; Xc= ? @ Yc[ Zc0c1cn 2c3c4c5c] 6c7c8c9c!c#c$c%c'c(c)c*c+c,c-c.c/c:c;c=c?c@c[c]c^c_c`c{c|c}ce ~cadf bdcdg ddedo fdgdhdidjdkdldmdndodpdqdrdsdtdudvdp wdxdq ydzdh AdBdi CdDdj EdFdGdHdr IdJdKdLdMdNds OdPdt QdRdu SdTdv UdVdw WdXdYdZdx 0d1d2d3dy 4d5d6d7d8d9d!d#d$d%d'd(d)d*d+d^ _ ` ,d-d.d/d:d;d=d?d@d[d]d^d_d`d{d|d}d~daebece{ deeefegeheiejekelemeneoepeqereseteuevewexeyezeAeBeCeDeEeFeGeHeIeJez A | } ~ KeLeabMeNeOebbcbPeQeReSeTeUeVeWeXeYeZe0e1e2e3e4e5e6e7e8e9e!e#e$e%e'e(e)e*e+e,e-e.e/e:e;e=e?e@e[e]e^e_e`e{e|e}e~eafbfcfdfefffgfhfifjfkflfmfnfofpfqfrfsftfufvfdbebfbgbhbibjbkblbmbnbobpbqbrbsbtbubvbwbxbybzbAbBbCbDbEbFbGbHbIbJbKbwfxfyfzfAfBfCfDfEfFfGfHfIfJfKfLfMfNfOfPfQfRfSfTfUfB VfC WfXfD YfZf0f1f2f3f4f5fE F 6f7f8f9f!fG H I J K L M N #f$f%f'f(f)f*f+f,f-f.f/f:f;f=f?f@f[f]f^f_f`f{f|f}f~fagbgcgdgegfggghgigjgkglgmgngogpgqgrgsgtgugvgwgxgygzgAgBgCgDgEgFgGgHgIgJgKgLgMgNgOgPgQgRgSgTgUgVgWgXgYg

63 if singleton is None: 582 ctx2a NbObPbQbRbSbTbUbVbWbXbYbZb0b1b2bO P Q R S T U V W X Y 3b4b5b6b7b8b9b!b#b$b%b'b(b)b*b+b,b-b.b/b:b;b=b?b@b[b]b^b_b`b{b|b}b~bacZ 0 1 2 3 4 bcccdcecfcgc5 hc6 ic7 8 jckclc9 mc! ncocpcqcrcsctcucvcwc# xcyc$ zcAcBcCc% ' Dc( ) * + Ec, - . Fck Gc/ m HcIcJc: KcLcd c McNcOcPcl QcRcScTcUcVcWc; Xc= ? @ Yc[ Zc0c1cn 2c3c4c5c] 6c7c8c9c!c#c$c%c'c(c)c*c+c,c-c.c/c:c;c=c?c@c[c]c^c_c`c{c|c}ce ~cadf bdcdg ddedo fdgdhdidjdkdldmdndodpdqdrdsdtdudvdp wdxdq ydzdh AdBdi CdDdj EdFdGdHdr IdJdKdLdMdNds OdPdt QdRdu SdTdv UdVdw WdXdYdZdx 0d1d2d3dy 4d5d6d7d8d9d!d#d$d%d'd(d)d*d+d^ _ ` ,d-d.d/d:d;d=d?d@d[d]d^d_d`d{d|d}d~daebece{ deeefegeheiejekelemeneoepeqereseteuevewexeyezeAeBeCeDeEeFeGeHeIeJez A | } ~ KeLeabMeNeOebbcbPeQeReSeTeUeVeWeXeYeZe0e1e2e3e4e5e6e7e8e9e!e#e$e%e'e(e)e*e+e,e-e.e/e:e;e=e?e@e[e]e^e_e`e{e|e}e~eafbfcfdfefffgfhfifjfkflfmfnfofpfqfrfsftfufvfdbebfbgbhbibjbkblbmbnbobpbqbrbsbtbubvbwbxbybzbAbBbCbDbEbFbGbHbIbJbKbwfxfyfzfAfBfCfDfEfFfGfHfIfJfKfLfMfNfOfPfQfRfSfTfUfB VfC WfXfD YfZf0f1f2f3f4f5fE F 6f7f8f9f!fG H I J K L M N #f$f%f'f(f)f*f+f,f-f.f/f:f;f=f?f@f[f]f^f_f`f{f|f}f~fagbgcgdgegfggghgigjgkglgmgngogpgqgrgsgtgugvgwgxgygzgAgBgCgDgEgFgGgHgIgJgKgLgMgNgOgPgQgRgSgTgUgVgWgXgYg

64 raise ValueError(f"{value} is not a valid {cls.__name__}") 3 ctx1cln

65 return singleton 582 ctx2a NbObPbQbRbSbTbUbVbWbXbYbZb0b1b2bO P Q R S T U V W X Y 3b4b5b6b7b8b9b!b#b$b%b'b(b)b*b+b,b-b.b/b:b;b=b?b@b[b]b^b_b`b{b|b}b~bacZ 0 1 2 3 4 bcccdcecfcgc5 hc6 ic7 8 jckclc9 mc! ncocpcqcrcsctcucvcwc# xcyc$ zcAcBcCc% ' Dc( ) * + Ec, - . Fck Gc/ m HcIcJc: KcLcd c McNcOcPcl QcRcScTcUcVcWc; Xc= ? @ Yc[ Zc0c1cn 2c3c4c5c] 6c7c8c9c!c#c$c%c'c(c)c*c+c,c-c.c/c:c;c=c?c@c[c]c^c_c`c{c|c}ce ~cadf bdcdg ddedo fdgdhdidjdkdldmdndodpdqdrdsdtdudvdp wdxdq ydzdh AdBdi CdDdj EdFdGdHdr IdJdKdLdMdNds OdPdt QdRdu SdTdv UdVdw WdXdYdZdx 0d1d2d3dy 4d5d6d7d8d9d!d#d$d%d'd(d)d*d+d^ _ ` ,d-d.d/d:d;d=d?d@d[d]d^d_d`d{d|d}d~daebece{ deeefegeheiejekelemeneoepeqereseteuevewexeyezeAeBeCeDeEeFeGeHeIeJez A | } ~ KeLeabMeNeOebbcbPeQeReSeTeUeVeWeXeYeZe0e1e2e3e4e5e6e7e8e9e!e#e$e%e'e(e)e*e+e,e-e.e/e:e;e=e?e@e[e]e^e_e`e{e|e}e~eafbfcfdfefffgfhfifjfkflfmfnfofpfqfrfsftfufvfdbebfbgbhbibjbkblbmbnbobpbqbrbsbtbubvbwbxbybzbAbBbCbDbEbFbGbHbIbJbKbwfxfyfzfAfBfCfDfEfFfGfHfIfJfKfLfMfNfOfPfQfRfSfTfUfB VfC WfXfD YfZf0f1f2f3f4f5fE F 6f7f8f9f!fG H I J K L M N #f$f%f'f(f)f*f+f,f-f.f/f:f;f=f?f@f[f]f^f_f`f{f|f}f~fagbgcgdgegfggghgigjgkglgmgngogpgqgrgsgtgugvgwgxgygzgAgBgCgDgEgFgGgHgIgJgKgLgMgNgOgPgQgRgSgTgUgVgWgXgYg

66 

67 def __repr__(self) -> str: (empty)

68 return f"<{self.__class__.__name__}.{self._name}: {int(self)}>" 12 ctx2a c n LbMbe f g h i j 2g

69 

70 @property (empty)

71 def name(self) -> str: (empty)

72 return self._name 55 ctx156789!#$%'()*+,-.k/m:dcefgopqhijrstuvwxyzABCDEFGHIJKLMN

73 

74 @property (empty)

75 def value(self) -> int: (empty)

76 return int(self) 112 ctx2O P Q R S T U V W X Y 3g4g5g6g7g8g9g!g#g$gZ 0 1 2 3 4 m c ; = ? @ [ %g] e f g o p q h i j r s t u v w x y ^ _ ` { z A | } ~ abbbcbdbebfbgbhbibjbkblbmbnbobpbqbrbsbtbubvbwbxbybzbAbBbCbDbEbFbGbHbIbJbKbB C D E F G H I J K L M N