22 lines
513 B
Go
22 lines
513 B
Go
|
//go:build linux || windows
|
||
|
|
||
|
package gpu
|
||
|
|
||
|
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, ",")
|
||
|
}
|