diff --git a/pkg/nvml/lib.go b/pkg/nvml/lib.go index 4d26531..bc4c3de 100644 --- a/pkg/nvml/lib.go +++ b/pkg/nvml/lib.go @@ -198,93 +198,93 @@ func (pis ProcessInfo_v2Slice) ToProcessInfoSlice() []ProcessInfo { // When new versioned symbols are added, these would have to be initialized above and have // corresponding checks and subsequent assignments added below. func (l *library) updateVersionedSymbols() { - err := l.LookupSymbol("nvmlInit_v2") + err := l.dl.Lookup("nvmlInit_v2") if err == nil { nvmlInit = nvmlInit_v2 } - err = l.LookupSymbol("nvmlDeviceGetPciInfo_v2") + err = l.dl.Lookup("nvmlDeviceGetPciInfo_v2") if err == nil { nvmlDeviceGetPciInfo = nvmlDeviceGetPciInfo_v2 } - err = l.LookupSymbol("nvmlDeviceGetPciInfo_v3") + err = l.dl.Lookup("nvmlDeviceGetPciInfo_v3") if err == nil { nvmlDeviceGetPciInfo = nvmlDeviceGetPciInfo_v3 } - err = l.LookupSymbol("nvmlDeviceGetCount_v2") + err = l.dl.Lookup("nvmlDeviceGetCount_v2") if err == nil { nvmlDeviceGetCount = nvmlDeviceGetCount_v2 } - err = l.LookupSymbol("nvmlDeviceGetHandleByIndex_v2") + err = l.dl.Lookup("nvmlDeviceGetHandleByIndex_v2") if err == nil { nvmlDeviceGetHandleByIndex = nvmlDeviceGetHandleByIndex_v2 } - err = l.LookupSymbol("nvmlDeviceGetHandleByPciBusId_v2") + err = l.dl.Lookup("nvmlDeviceGetHandleByPciBusId_v2") if err == nil { nvmlDeviceGetHandleByPciBusId = nvmlDeviceGetHandleByPciBusId_v2 } - err = l.LookupSymbol("nvmlDeviceGetNvLinkRemotePciInfo_v2") + err = l.dl.Lookup("nvmlDeviceGetNvLinkRemotePciInfo_v2") if err == nil { nvmlDeviceGetNvLinkRemotePciInfo = nvmlDeviceGetNvLinkRemotePciInfo_v2 } // Unable to overwrite nvmlDeviceRemoveGpu() because the v2 function takes // a different set of parameters than the v1 function. - //err = l.LookupSymbol("nvmlDeviceRemoveGpu_v2") + //err = l.dl.Lookup("nvmlDeviceRemoveGpu_v2") //if err == nil { // nvmlDeviceRemoveGpu = nvmlDeviceRemoveGpu_v2 //} - err = l.LookupSymbol("nvmlDeviceGetGridLicensableFeatures_v2") + err = l.dl.Lookup("nvmlDeviceGetGridLicensableFeatures_v2") if err == nil { nvmlDeviceGetGridLicensableFeatures = nvmlDeviceGetGridLicensableFeatures_v2 } - err = l.LookupSymbol("nvmlDeviceGetGridLicensableFeatures_v3") + err = l.dl.Lookup("nvmlDeviceGetGridLicensableFeatures_v3") if err == nil { nvmlDeviceGetGridLicensableFeatures = nvmlDeviceGetGridLicensableFeatures_v3 } - err = l.LookupSymbol("nvmlDeviceGetGridLicensableFeatures_v4") + err = l.dl.Lookup("nvmlDeviceGetGridLicensableFeatures_v4") if err == nil { nvmlDeviceGetGridLicensableFeatures = nvmlDeviceGetGridLicensableFeatures_v4 } - err = l.LookupSymbol("nvmlEventSetWait_v2") + err = l.dl.Lookup("nvmlEventSetWait_v2") if err == nil { nvmlEventSetWait = nvmlEventSetWait_v2 } - err = l.LookupSymbol("nvmlDeviceGetAttributes_v2") + err = l.dl.Lookup("nvmlDeviceGetAttributes_v2") if err == nil { nvmlDeviceGetAttributes = nvmlDeviceGetAttributes_v2 } - err = l.LookupSymbol("nvmlComputeInstanceGetInfo_v2") + err = l.dl.Lookup("nvmlComputeInstanceGetInfo_v2") if err == nil { nvmlComputeInstanceGetInfo = nvmlComputeInstanceGetInfo_v2 } - err = l.LookupSymbol("nvmlDeviceGetComputeRunningProcesses_v2") + err = l.dl.Lookup("nvmlDeviceGetComputeRunningProcesses_v2") if err == nil { deviceGetComputeRunningProcesses = deviceGetComputeRunningProcesses_v2 } - err = l.LookupSymbol("nvmlDeviceGetComputeRunningProcesses_v3") + err = l.dl.Lookup("nvmlDeviceGetComputeRunningProcesses_v3") if err == nil { deviceGetComputeRunningProcesses = deviceGetComputeRunningProcesses_v3 } - err = l.LookupSymbol("nvmlDeviceGetGraphicsRunningProcesses_v2") + err = l.dl.Lookup("nvmlDeviceGetGraphicsRunningProcesses_v2") if err == nil { deviceGetGraphicsRunningProcesses = deviceGetGraphicsRunningProcesses_v2 } - err = l.LookupSymbol("nvmlDeviceGetGraphicsRunningProcesses_v3") + err = l.dl.Lookup("nvmlDeviceGetGraphicsRunningProcesses_v3") if err == nil { deviceGetGraphicsRunningProcesses = deviceGetGraphicsRunningProcesses_v3 } - err = l.LookupSymbol("nvmlDeviceGetMPSComputeRunningProcesses_v2") + err = l.dl.Lookup("nvmlDeviceGetMPSComputeRunningProcesses_v2") if err == nil { deviceGetMPSComputeRunningProcesses = deviceGetMPSComputeRunningProcesses_v2 } - err = l.LookupSymbol("nvmlDeviceGetMPSComputeRunningProcesses_v3") + err = l.dl.Lookup("nvmlDeviceGetMPSComputeRunningProcesses_v3") if err == nil { deviceGetMPSComputeRunningProcesses = deviceGetMPSComputeRunningProcesses_v3 } - err = l.LookupSymbol("nvmlDeviceGetGpuInstancePossiblePlacements_v2") + err = l.dl.Lookup("nvmlDeviceGetGpuInstancePossiblePlacements_v2") if err == nil { nvmlDeviceGetGpuInstancePossiblePlacements = nvmlDeviceGetGpuInstancePossiblePlacements_v2 } - err = l.LookupSymbol("nvmlVgpuInstanceGetLicenseInfo_v2") + err = l.dl.Lookup("nvmlVgpuInstanceGetLicenseInfo_v2") if err == nil { nvmlVgpuInstanceGetLicenseInfo = nvmlVgpuInstanceGetLicenseInfo_v2 }