Add wrappers for NvLink NVML functions

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

Closed rjzamora closed 5 years ago

rjzamora commented 5 years ago

This PR addresses issue#8. Adding wrappers (and corresponding tests) for the 11 NvLink methods available in NVML.

mrocklin commented 5 years ago

@rjzamora things here look fine to me. I'm not sure I'm particularly able to judge here though. If it works it works I suppose.

rjzamora commented 5 years ago

Good point :) I'll do some benchmark testing and merge if everything looks "correct"