From 4780568e967754176e8e969cfcb93b2a0245418e Mon Sep 17 00:00:00 2001 From: Douglas Wightman Date: Wed, 22 Oct 2025 10:28:47 -0600 Subject: [PATCH] Update max entity count to DCGM_GROUP_MAX_ENTITIES_V2 --- pkg/dcgm/device_info.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pkg/dcgm/device_info.go b/pkg/dcgm/device_info.go index 71c3b1b..92f01b4 100644 --- a/pkg/dcgm/device_info.go +++ b/pkg/dcgm/device_info.go @@ -69,8 +69,8 @@ func getAllDeviceCount() (gpuCount uint, err error) { // getAllDeviceCount counts all GPUs on the system func getEntityGroupEntities(entityGroup Field_Entity_Group) ([]uint, error) { var err error - var pEntities [C.DCGM_MAX_NUM_DEVICES]C.uint - var count C.int = C.DCGM_MAX_NUM_DEVICES + var pEntities [C.DCGM_GROUP_MAX_ENTITIES_V2]C.uint + var count C.int = C.DCGM_GROUP_MAX_ENTITIES_V2 result := C.dcgmGetEntityGroupEntities(handle.handle, C.dcgm_field_entity_group_t(entityGroup), &pEntities[0], &count, 0) if err = errorString(result); err != nil {