Skip to content

Commit 3af261a

Browse files
committed
add the system class
1 parent f1b1c4e commit 3af261a

File tree

1 file changed

+7
-13
lines changed

1 file changed

+7
-13
lines changed

cuda_core/cuda/core/experimental/_system.py

Lines changed: 7 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -6,10 +6,6 @@
66
class System:
77
"""Represent the system and provide information about the CUDA environment."""
88

9-
_instance = None
10-
_driver_version = None
11-
_devices = None
12-
139
def __new__(cls):
1410
if not cls._instance:
1511
cls._instance = super(System, cls).__new__(cls)
@@ -24,13 +20,12 @@ def driver_version(self) -> Tuple[int, int]:
2420
"""Query the CUDA driver version.
2521
2622
Returns:
27-
namedtuple: a 2-tuple of (major, minor).
23+
tuple: a 2-tuple of (major, minor).
2824
"""
29-
if self._driver_version is None:
30-
version = handle_return(cuda.cuDriverGetVersion())
31-
major = version // 1000
32-
minor = (version % 1000) // 10
33-
self._driver_version = (major, minor)
25+
version = handle_return(cuda.cuDriverGetVersion())
26+
major = version // 1000
27+
minor = (version % 1000) // 10
28+
self._driver_version = (major, minor)
3429
return self._driver_version
3530

3631
@property
@@ -41,9 +36,8 @@ def num_devices(self) -> int:
4136
@property
4237
def devices(self) -> tuple:
4338
"""Query the available device instances."""
44-
if self._devices is None:
45-
total = self.num_devices
46-
self._devices = tuple(Device(device_id) for device_id in range(total))
39+
total = self.num_devices
40+
self._devices = tuple(Device(device_id) for device_id in range(total))
4741
return self._devices
4842

4943
_system = System.__new__(System)

0 commit comments

Comments
 (0)