Method SelectGPU

vgpu/gpu.go:333–438

(gpus []vk.PhysicalDevice, gpuCount int)

}

func (gp *GPU) SelectGPU(gpus []vk.PhysicalDevice, gpuCount int) int {
	if gpuCount == 1 {
		var properties vk.PhysicalDeviceProperties
		vk.GetPhysicalDeviceProperties(gpus[0], &properties)
		properties.Deref()
		gp.DeviceName = gp.GetDeviceName(&properties, 0)
		if Debug {
			log.Printf("vgpu: selected only device named: %s\n", gp.DeviceName)
		}
		return 0
	}
	trgDevNm := ""
	if ev := os.Getenv("MESA_VK_DEVICE_SELECT"); ev != "" {
		trgDevNm = ev
	} else if ev := os.Getenv("VK_DEVICE_SELECT"); ev != "" {
		trgDevNm = ev
	}
	if gp.Compute {
		if ev := os.Getenv("VK_COMPUTE_DEVICE_SELECT"); ev != "" {
			trgDevNm = ev
		}
	}

	if trgDevNm != "" {
		idx, err := strconv.Atoi(trgDevNm)
		if err == nil && idx >= 0 && idx < gpuCount {
			curIndex := 0
			for gi := 0; gi < gpuCount; gi++ {
				var properties vk.PhysicalDeviceProperties
				vk.GetPhysicalDeviceProperties(gpus[gi], &properties)
				properties.Deref()
				if properties.DeviceType == vk.PhysicalDeviceTypeDiscreteGpu {
					if curIndex == idx {
						gp.DeviceName = gp.GetDeviceName(&properties, gi)
						if Debug {
							log.Printf("vgpu: selected device named: %s, specified by index in *_DEVICE_SELECT environment variable, index: %d\n", gp.DeviceName, gi)
						}
						return gi
					} else {
						curIndex++
					}
				}
			}
			panic(fmt.Sprintf("vgpu: device specified by index in *_DEVICE_SELECT environment variable, index: %d, NOT FOUND\n", idx))
		}
		for gi := 0; gi < gpuCount; gi++ {
			var properties vk.PhysicalDeviceProperties
			vk.GetPhysicalDeviceProperties(gpus[gi], &properties)
			properties.Deref()
			if bytes.Contains(properties.DeviceName[:], []byte(trgDevNm)) {
				devNm := gp.GetDeviceName(&properties, gi)
				if Debug {
					log.Printf("vgpu: selected device named: %s, specified in *_DEVICE_SELECT environment variable, index: %d\n", devNm, gi)
				}
				gp.DeviceName = devNm
				return gi
			}
		}