From 6c243c3a8d1b102cfef0b139c3590993d8775dc3 Mon Sep 17 00:00:00 2001 From: Seth Hoenig Date: Tue, 20 Aug 2024 19:33:50 +0000 Subject: [PATCH] wip: do not use _v2 functions --- nvml/driver_linux.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nvml/driver_linux.go b/nvml/driver_linux.go index 91b5d97..8673b41 100644 --- a/nvml/driver_linux.go +++ b/nvml/driver_linux.go @@ -110,7 +110,7 @@ func (n *nvmlDriver) DeviceInfoByUUID(uuid string) (*DeviceInfo, error) { return nil, decode("failed to get device name", code) } - memory, code := nvml.DeviceGetMemoryInfo_v2(device) + memory, code := nvml.DeviceGetMemoryInfo(device) if code != nvml.SUCCESS { return nil, decode("failed to get device memory info", code) } @@ -238,7 +238,7 @@ func (n *nvmlDriver) DeviceInfoAndStatusByUUID(uuid string) (*DeviceInfo, *Devic return nil, nil, decode("failed to get device info", code) } - mem, code := nvml.DeviceGetMemoryInfo_v2(device) + mem, code := nvml.DeviceGetMemoryInfo(device) if code != nvml.SUCCESS { return nil, nil, decode("failed to get device memory utilization", code) }