func()

in pkg/gpu/nvidia/mig/mig.go [250:268]

• 17 lines of code
• 5 McCabe index (conditional complexity)

func (d *DeviceManager) discoverNumGPUs() (int, error) {
	numGPUs := 0

	reg := regexp.MustCompile(nvidiaDeviceRE)
	files, err := ioutil.ReadDir(d.devDirectory)
	if err != nil {
		return 0, fmt.Errorf("failed to read devices on node: %v", err)
	}

	for _, f := range files {
		if f.IsDir() {
			continue
		}
		if reg.MatchString(f.Name()) {
			numGPUs++
		}
	}
	return numGPUs, nil
}