gpuopenanalytics / pynvml

Provide Python access to the NVML library for GPU diagnostics
BSD 3-Clause "New" or "Revised" License
205 stars 31 forks source link

Fix nvml enum bug and update pytests #35

Closed rjzamora closed 3 years ago

rjzamora commented 3 years ago

Follow-up to #33 and #34 - Fixes a minor bugs in the _nvmlGpuP2PCapsIndex_t definition and in test_nvmlSystemGetCudaDriverVersion. Also introduces MIG fixtures and updates formatting in the test suite with black.