Skip to content

Commit ce333b6

Browse files
mdboomCopilot
andauthored
cuda.core.system: affinity, clock, fan, temperature and thermals (#1492)
* cuda.core.system: More device-related APIs * Fix line wrapping * Hide handle as an implementation detail * Update cuda_core/cuda/core/system/_device.pyx Co-authored-by: Copilot <[email protected]> * Address comments from Copilot * Working on tests * Fix tests * Fix creating new device * Fix iterator * Fix affinity test * Fix nearest GPUs * cuda.core.system: affinity, clock, fan, temperature and thermals * Address comments from Copilot * Fix tests * Fix test on WDDM Windows * Add comments about pointer lifetime. * Fix tests --------- Co-authored-by: Copilot <[email protected]>
1 parent 1af5db9 commit ce333b6

File tree

11 files changed

+1142
-387
lines changed

11 files changed

+1142
-387
lines changed

cuda_bindings/cuda/bindings/_nvml.pxd

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@ from .cy_nvml cimport *
1414
###############################################################################
1515

1616
ctypedef nvmlDramEncryptionInfo_v1_t DramEncryptionInfo_v1
17+
ctypedef nvmlMarginTemperature_v1_t MarginTemperature_v1
18+
ctypedef nvmlFanSpeedInfo_v1_t FanSpeedInfo_v1
1719
ctypedef nvmlConfComputeSetKeyRotationThresholdInfo_v1_t ConfComputeSetKeyRotationThresholdInfo_v1
1820
ctypedef nvmlSystemDriverBranchInfo_v1_t SystemDriverBranchInfo_v1
1921
ctypedef nvmlTemperature_v1_t Temperature_v1
@@ -196,14 +198,12 @@ cpdef object device_get_supported_graphics_clocks(intptr_t device, unsigned int
196198
cpdef tuple device_get_auto_boosted_clocks_enabled(intptr_t device)
197199
cpdef unsigned int device_get_fan_speed(intptr_t device) except? 0
198200
cpdef unsigned int device_get_fan_speed_v2(intptr_t device, unsigned int fan) except? 0
199-
cpdef object device_get_fan_speed_rpm(intptr_t device)
200201
cpdef unsigned int device_get_target_fan_speed(intptr_t device, unsigned int fan) except? 0
201202
cpdef tuple device_get_min_max_fan_speed(intptr_t device)
202203
cpdef unsigned int device_get_fan_control_policy_v2(intptr_t device, unsigned int fan) except *
203204
cpdef unsigned int device_get_num_fans(intptr_t device) except? 0
204205
cpdef object device_get_cooler_info(intptr_t device)
205206
cpdef unsigned int device_get_temperature_threshold(intptr_t device, int threshold_type) except? 0
206-
cpdef object device_get_margin_temperature(intptr_t device)
207207
cpdef object device_get_thermal_settings(intptr_t device, unsigned int sensor_ind_ex)
208208
cpdef int device_get_performance_state(intptr_t device) except? -1
209209
cpdef unsigned long long device_get_current_clocks_event_reasons(intptr_t device) except? 0
@@ -214,7 +214,6 @@ cpdef int device_get_mem_clk_vf_offset(intptr_t device) except? 0
214214
cpdef tuple device_get_min_max_clock_of_p_state(intptr_t device, int type, int pstate)
215215
cpdef tuple device_get_gpc_clk_min_max_vf_offset(intptr_t device)
216216
cpdef tuple device_get_mem_clk_min_max_vf_offset(intptr_t device)
217-
cpdef object device_get_clock_offsets(intptr_t device)
218217
cpdef device_set_clock_offsets(intptr_t device, intptr_t info)
219218
cpdef object device_get_performance_modes(intptr_t device)
220219
cpdef object device_get_current_clock_freqs(intptr_t device)

0 commit comments

Comments
 (0)