Add ability to create an NVML Device from a UUID

Signed-off-by: Kevin Klues <[email protected]>

bindings/go/nvml/bindings.go

@@ -227,6 +227,13 @@ func deviceGetHandleByIndex(idx uint) (handle, error) {
 	return handle{dev}, errorString(r)
 }
 
+func deviceGetHandleByUUID(uuid string) (handle, error) {
+	var dev C.nvmlDevice_t
+
+	r := C.nvmlDeviceGetHandleByUUID(C.CString(uuid), &dev)
+	return handle{dev}, errorString(r)
+}
+
 func deviceGetTopologyCommonAncestor(h1, h2 handle) (*uint, error) {
 	r := dl.lookupSymbol("nvmlDeviceGetTopologyCommonAncestor")
 	if r == C.NVML_ERROR_FUNCTION_NOT_FOUND {

bindings/go/nvml/nvml.go

@@ -336,6 +336,30 @@ func NewDevice(idx uint) (device *Device, err error) {
 
 	h, err := deviceGetHandleByIndex(idx)
 	assert(err)
+
+	device, err = newDevice(h)
+	assert(err)
+
+	return device, err
+}
+
+func NewDeviceByUUID(uuid string) (device *Device, err error) {
+	defer func() {
+		if r := recover(); r != nil {
+			err = r.(error)
+		}
+	}()
+
+	h, err := deviceGetHandleByUUID(uuid)
+	assert(err)
+
+	device, err = newDevice(h)
+	assert(err)
+
+	return device, err
+}
+
+func newDevice(h handle) (device *Device, err error) {
 	model, err := h.deviceGetName()
 	assert(err)
 	uuid, err := h.deviceGetUUID()
@@ -413,6 +437,30 @@ func NewDeviceLite(idx uint) (device *Device, err error) {
 
 	h, err := deviceGetHandleByIndex(idx)
 	assert(err)
+
+	device, err = newDeviceLite(h)
+	assert(err)
+
+	return device, err
+}
+
+func NewDeviceLiteByUUID(uuid string) (device *Device, err error) {
+	defer func() {
+		if r := recover(); r != nil {
+			err = r.(error)
+		}
+	}()
+
+	h, err := deviceGetHandleByUUID(uuid)
+	assert(err)
+
+	device, err = newDeviceLite(h)
+	assert(err)
+
+	return device, err
+}
+
+func newDeviceLite(h handle) (device *Device, err error) {
 	uuid, err := h.deviceGetUUID()
 	assert(err)
 	minor, err := h.deviceGetMinorNumber()