how to get the same result `nvidia-smi nvlink -s -i 0` by using go-nvml

[yxxhero]

how to get the same result nvidia-smi nvlink -s -i 0 by using go-nvml

thanks so much in advance.

[jiayelamazon]

Any updates on this?

[yxxhero]

no.

[yxxhero]

ping

[yxxhero]

package main
 
import (
    "fmt"
    "github.com/NVIDIA/go-nvml/pkg/nvml"
)
 
func main() {

    ret := nvml.Init()
    if ret != nvml.SUCCESS {
        fmt.Printf("Failed to initialize NVML: %s\n", nvml.ErrorString(ret))
        return
    }
    defer nvml.Shutdown()
 
   
    device, ret := nvml.DeviceGetHandleByIndex(0)
    if ret != nvml.SUCCESS {
        fmt.Printf("Failed to get device handle: %s\n", nvml.ErrorString(ret))
        return
    }
 
 
    for link := 0; link < nvml.NVLINK_MAX_LINKS; link++ {
        state, ret := device.GetNvLinkState(link)
        if ret != nvml.SUCCESS {
            continue
        }
 
        if state == nvml.FEATURE_ENABLED {
            fmt.Printf("NVLink %d:\n", link)
            
 
            version, _ := device.GetNvLinkVersion(link)
            fmt.Printf("  Version: %d\n", version)
            
   
            p2pCap, _ := device.GetNvLinkCapability(link, nvml.NVLINK_CAP_P2P_SUPPORTED)
            fmt.Printf("  P2P Supported: %t\n", p2pCap == 1)

            pciInfo, _ := device.GetNvLinkRemotePciInfo(link)
            fmt.Printf("  Remote Device: %04X:%02X:%02X.%X\n", 
                pciInfo.Domain, pciInfo.Bus, pciInfo.Device, pciInfo.Function)
            

            for counter := 0; counter < 2; counter++ {
                rx, tx, _ := device.GetNvLinkUtilizationCounter(link, counter)
                fmt.Printf("  Counter %d: RX=%d, TX=%d\n", counter, rx, tx)
            }
            
      
            replayErrors, _ := device.GetNvLinkErrorCounter(link, nvml.NVLINK_ERROR_DL_REPLAY)
            fmt.Printf("  Replay Errors: %d\n", replayErrors)
            
            fmt.Println()
        }
    }
}

[yxxhero]

@jiayelamazon

[karthikvetrivel]

Hi @yxxhero — thanks for the question (and answer). Yep, the code shared in this thread covers the right approach: iterate over NVLink indices using device.GetNvLinkState(), then query per-link info with GetNvLinkVersion(), GetNvLinkCapability(), GetNvLinkRemotePciInfo(), GetNvLinkUtilizationCounter(), and GetNvLinkErrorCounter(). That should give you the equivalent of nvidia-smi nvlink -s -i 0.

Closing as this has been answered — feel free to reopen or file a new issue if you need further help!