persistence_mode -> is_persistence_mode_enabled

Author: mdboom

cuda_core/cuda/core/system/_device.pyx

@@ -222,16 +222,16 @@ cdef class Device:
         return bool(nvml.device_get_c2c_mode_info_v(self._handle).is_c2c_enabled)
 
     @property
-    def persistence_mode(self) -> bool:
+    def is_persistence_mode_enabled(self) -> bool:
         """
         Whether persistence mode is enabled for this device.
 
         For Linux only.
         """
         return nvml.device_get_persistence_mode(self._handle) == nvml.EnableState.FEATURE_ENABLED
 
-    @persistence_mode.setter
-    def persistence_mode(self, enabled: bool) -> None:
+    @is_persistence_mode_enabled.setter
+    def is_persistence_mode_enabled(self, enabled: bool) -> None:
         nvml.device_set_persistence_mode(
             self._handle,
             nvml.EnableState.FEATURE_ENABLED if enabled else nvml.EnableState.FEATURE_DISABLED

cuda_core/tests/system/test_system_device.py

@@ -343,16 +343,16 @@ def test_c2c_mode_enabled():
 @pytest.mark.skipif(helpers.IS_WSL or helpers.IS_WINDOWS, reason="Persistence mode not supported on WSL or Windows")
 def test_persistence_mode_enabled():
     for device in system.Device.get_all_devices():
-        is_enabled = device.persistence_mode
+        is_enabled = device.is_persistence_mode_enabled
         assert isinstance(is_enabled, bool)
         try:
-            device.persistence_mode = False
+            device.is_persistence_mode_enabled = False
         except nvml.NoPermissionError as e:
             pytest.xfail(f"nvml.NoPermissionError: {e}")
         try:
-            assert device.persistence_mode is False
+            assert device.is_persistence_mode_enabled is False
         finally:
-            device.persistence_mode = is_enabled
+            device.is_persistence_mode_enabled = is_enabled
 
 
 def test_field_values():