lotus/lib/harmony/resources/getGPU.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
}