35 lines
735 B
Go
35 lines
735 B
Go
//go:build !darwin
|
|
// +build !darwin
|
|
|
|
package resources
|
|
|
|
import (
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
ffi "github.com/filecoin-project/filecoin-ffi"
|
|
)
|
|
|
|
func getGPUDevices() float64 { // GPU boolean
|
|
if nstr := os.Getenv("HARMONY_OVERRIDE_GPUS"); nstr != "" {
|
|
n, err := strconv.ParseFloat(nstr, 64)
|
|
if err != nil {
|
|
logger.Errorf("parsing HARMONY_OVERRIDE_GPUS failed: %+v", err)
|
|
} else {
|
|
return n
|
|
}
|
|
}
|
|
|
|
gpus, err := ffi.GetGPUDevices()
|
|
logger.Infow("GPUs", "list", gpus)
|
|
if err != nil {
|
|
logger.Errorf("getting gpu devices failed: %+v", err)
|
|
}
|
|
all := strings.ToLower(strings.Join(gpus, ","))
|
|
if len(gpus) > 1 || strings.Contains(all, "ati") || strings.Contains(all, "nvidia") {
|
|
return float64(len(gpus))
|
|
}
|
|
return 0
|
|
}
|