diff --git a/bindings/go/nvml/nvml.go b/bindings/go/nvml/nvml.go index d3f1934..c7ab5a1 100644 --- a/bindings/go/nvml/nvml.go +++ b/bindings/go/nvml/nvml.go @@ -145,6 +145,12 @@ const ( FourNVLINKLinks FiveNVLINKLinks SixNVLINKLinks + SevenNVLINKLinks + EightNVLINKLinks + NineNVLINKLinks + TenNVLINKLinks + ElevenNVLINKLinks + TwelveNVLINKLinks ) type P2PLink struct { @@ -178,6 +184,18 @@ func (t P2PLinkType) String() string { return "Five NVLinks" case SixNVLINKLinks: return "Six NVLinks" + case SevenNVLINKLinks: + return "Seven NVLinks" + case EightNVLINKLinks: + return "Eight NVLinks" + case NineNVLINKLinks: + return "Nine NVLinks" + case TenNVLINKLinks: + return "Ten NVLinks" + case ElevenNVLINKLinks: + return "Eleven NVLinks" + case TwelveNVLINKLinks: + return "Twelve NVLinks" case P2PLinkUnknown: } return "N/A" @@ -618,6 +636,18 @@ func GetNVLink(dev1, dev2 *Device) (link P2PLinkType, err error) { nvlink = FiveNVLINKLinks case FiveNVLINKLinks: nvlink = SixNVLINKLinks + case SixNVLINKLinks: + nvlink = SevenNVLINKLinks + case SevenNVLINKLinks: + nvlink = EightNVLINKLinks + case EightNVLINKLinks: + nvlink = NineNVLINKLinks + case NineNVLINKLinks: + nvlink = TenNVLINKLinks + case TenNVLINKLinks: + nvlink = ElevenNVLINKLinks + case ElevenNVLINKLinks: + nvlink = TwelveNVLINKLinks } } }