ollama/discover/gpu_oneapi.go

//go:build linux || windows

package discover

import (
	"log/slog"
	"strings"
)

func oneapiGetVisibleDevicesEnv(gpuInfo []GpuInfo) (string, string) {
	ids := []string{}
	for _, info := range gpuInfo {
		if info.Library != "oneapi" {
			// TODO shouldn't happen if things are wired correctly...
			slog.Debug("oneapiGetVisibleDevicesEnv skipping over non-sycl device", "library", info.Library)
			continue
		}
		ids = append(ids, info.ID)
	}
	return "ONEAPI_DEVICE_SELECTOR", "level_zero:" + strings.Join(ids, ",")
}