diff --git a/pkg/dcgm/internal.go b/pkg/dcgm/internal.go index b6e1033..cfe8172 100644 --- a/pkg/dcgm/internal.go +++ b/pkg/dcgm/internal.go @@ -100,7 +100,7 @@ func InjectFieldValue(gpu uint, fieldID Short, fieldType uint, status int, ts in fieldId: C.ushort(fieldID), fieldType: C.ushort(fieldType), status: C.int(status), - ts: C.long(ts), + ts: C.int64_t(ts), } switch fieldType { diff --git a/pkg/dcgm/policy.go b/pkg/dcgm/policy.go index 466481a..019829c 100644 --- a/pkg/dcgm/policy.go +++ b/pkg/dcgm/policy.go @@ -366,7 +366,7 @@ func registerPolicy(ctx context.Context, groupID GroupHandle, typ ...policyCondi return nil, err } - result := C.dcgmPolicyRegister_v2(handle.handle, groupID.handle, condition, C.fpRecvUpdates(C.violationNotify), C.ulong(0)) + result := C.dcgmPolicyRegister_v2(handle.handle, groupID.handle, condition, C.fpRecvUpdates(C.violationNotify), C.uint64_t(0)) if err = errorString(result); err != nil { return nil, &Error{msg: C.GoString(C.errorString(result)), Code: result}