mirror of
https://github.com/Xevion/vastly.git
synced 2025-12-05 23:16:48 -06:00
200 lines
7.5 KiB
Go
200 lines
7.5 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"math"
|
|
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
type ScoreReason struct {
|
|
Reason string
|
|
Offset float64
|
|
Value interface{}
|
|
IsMultiplier bool
|
|
}
|
|
|
|
type ScoredOffer struct {
|
|
Offer Offer
|
|
Score float64
|
|
Reasons []ScoreReason
|
|
Latency *int32
|
|
}
|
|
|
|
var (
|
|
sugar *zap.SugaredLogger
|
|
)
|
|
|
|
func init() {
|
|
logger, err := zap.NewDevelopment()
|
|
sugar = logger.Sugar()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func AddReason(reasonList []ScoreReason, reason string, offset float64, multiplier bool, value interface{}) []ScoreReason {
|
|
return append(reasonList, ScoreReason{Reason: reason, Offset: offset, IsMultiplier: multiplier, Value: fmt.Sprintf("%v", value)})
|
|
}
|
|
|
|
func ScoreOffers(offers []Offer) []ScoredOffer {
|
|
var scoredOffers = make([]ScoredOffer, 0, len(offers))
|
|
|
|
for _, offer := range offers {
|
|
reasons := make([]ScoreReason, 0, 16)
|
|
|
|
// Judge cost
|
|
targetCost := 0.42 // $/hour
|
|
costMultiplier := targetCost / offer.Search.TotalHour // e.x $0.42 / $0.50 = x0.84
|
|
// sugar.Debugw("Cost", "offer", offer.ID, "costMultiplier", costMultiplier, "targetCost", targetCost, "actualCost", offer.Search.TotalHour)
|
|
if math.Abs(costMultiplier-1.0) > 0.05 {
|
|
reasons = AddReason(reasons, "Cost Multiplier", costMultiplier, true, offer.Search.TotalHour)
|
|
}
|
|
|
|
// Judge DLPerf
|
|
targetDLPerf := 85.0
|
|
dlPerfMultiplier := offer.DLPerf / targetDLPerf // e.x 100 / 85 = x1.18
|
|
if math.Abs(dlPerfMultiplier-1.0) > 0.03 {
|
|
if dlPerfMultiplier > 1.0 {
|
|
dlPerfMultiplier = math.Sqrt(dlPerfMultiplier)
|
|
} else {
|
|
dlPerfMultiplier = math.Pow(dlPerfMultiplier, 2.0)
|
|
}
|
|
reasons = AddReason(reasons, "DLPerf Multiplier", dlPerfMultiplier, true, offer.DLPerf)
|
|
}
|
|
|
|
// Judge Internet Download Speed
|
|
if offer.InetDown < 150.0 {
|
|
reasons = AddReason(reasons, "Very Poor Internet Download Speed", -7.0, false, offer.InetDown)
|
|
} else if offer.InetDown < 300.0 {
|
|
reasons = AddReason(reasons, "Poor Internet Download Speed", -3.0, false, offer.InetDown)
|
|
} else if offer.InetDown < 500.0 {
|
|
reasons = AddReason(reasons, "Decent Internet Download Speed", 1.0, false, offer.InetDown)
|
|
} else if offer.InetDown < 1000.0 {
|
|
reasons = AddReason(reasons, "Good Internet Download Speed", 3.0, false, offer.InetDown)
|
|
} else if offer.InetDown >= 2000.0 {
|
|
reasons = AddReason(reasons, "Great Internet Download Speed", 4.0, false, offer.InetDown)
|
|
}
|
|
|
|
// Judge Internet Upload Speed
|
|
if offer.InetUp < 50.0 {
|
|
reasons = AddReason(reasons, "Extremely Poor Internet Upload Speed", -9.0, false, offer.InetUp)
|
|
} else if offer.InetUp < 100.0 {
|
|
reasons = AddReason(reasons, "Very Poor Internet Upload Speed", -5.0, false, offer.InetUp)
|
|
} else if offer.InetUp < 200.0 {
|
|
reasons = AddReason(reasons, "Poor Internet Upload Speed", -3.0, false, offer.InetUp)
|
|
} else if offer.InetUp < 400.0 {
|
|
reasons = AddReason(reasons, "Decent Internet Upload Speed", 1.0, false, offer.InetUp)
|
|
} else if offer.InetUp >= 800.0 {
|
|
reasons = AddReason(reasons, "Great Internet Upload Speed", 3.0, false, offer.InetUp)
|
|
} else if offer.InetUp >= 1000.0 {
|
|
reasons = AddReason(reasons, "Amazing Internet Upload Speed", 4.0, false, offer.InetUp)
|
|
}
|
|
|
|
// Judge verification
|
|
if offer.Verification == "verified" {
|
|
reasons = AddReason(reasons, "Verified", 2.0, false, offer.Verification)
|
|
} else {
|
|
reasons = AddReason(reasons, "Not Verified", -5.0, false, offer.Verification)
|
|
}
|
|
|
|
// Judge direct port count
|
|
if offer.DirectPortCount < 8 {
|
|
reasons = AddReason(reasons, "Very low direct port count", -2.0, false, offer.DirectPortCount)
|
|
} else if offer.DirectPortCount >= 32 {
|
|
reasons = AddReason(reasons, "Decent port count", 0.5, false, offer.DirectPortCount)
|
|
} else if offer.DirectPortCount >= 100 {
|
|
reasons = AddReason(reasons, "High port count", 1.0, false, offer.DirectPortCount)
|
|
}
|
|
|
|
// Judge CPU memory
|
|
if offer.CPURam < 16*1024 {
|
|
reasons = AddReason(reasons, "Low CPU memory", -4.0, false, offer.CPURam)
|
|
} else if offer.CPURam >= 31*1024 {
|
|
reasons = AddReason(reasons, "High CPU memory", 2.0, false, offer.CPURam)
|
|
} else if offer.CPURam >= 63*1024 {
|
|
reasons = AddReason(reasons, "Very High CPU memory", 3.0, false, offer.CPURam)
|
|
}
|
|
|
|
// Judge GPU count
|
|
if offer.NumGPUs < 1 {
|
|
reasons = AddReason(reasons, "No GPUs", -20.0, false, offer.NumGPUs)
|
|
} else if offer.NumGPUs == 2 {
|
|
reasons = AddReason(reasons, "Dual GPU", 1.0, false, offer.NumGPUs)
|
|
} else if offer.NumGPUs >= 3 {
|
|
reasons = AddReason(reasons, "Multi GPU", 2.0, false, offer.NumGPUs)
|
|
}
|
|
|
|
// Judge CUDA version
|
|
if offer.CudaMaxGood < 11.0 {
|
|
reasons = AddReason(reasons, "CUDA version very outdated", -7.0, false, offer.CudaMaxGood)
|
|
} else if offer.CudaMaxGood < 12.0 {
|
|
reasons = AddReason(reasons, "CUDA version outdated", -5.0, false, offer.CudaMaxGood)
|
|
} else if offer.CudaMaxGood >= 12.5 {
|
|
reasons = AddReason(reasons, "CUDA version high", 5.0, false, offer.CudaMaxGood)
|
|
} else if offer.CudaMaxGood >= 12.0 {
|
|
reasons = AddReason(reasons, "CUDA version decent", 1.5, false, offer.CudaMaxGood)
|
|
}
|
|
|
|
// Judge effective core count
|
|
if offer.CPUCoresEffective < 4.0 {
|
|
reasons = AddReason(reasons, "Low core count", -5.0, false, offer.CPUCoresEffective)
|
|
} else if offer.CPUCoresEffective >= 8.0 {
|
|
reasons = AddReason(reasons, "High core count", 5.0, false, offer.CPUCoresEffective)
|
|
} else if offer.CPUCoresEffective >= 6.0 {
|
|
reasons = AddReason(reasons, "Decent core count", 2.0, false, offer.CPUCoresEffective)
|
|
}
|
|
|
|
// Judge disk space available
|
|
if offer.DiskSpace < 100.0 {
|
|
reasons = AddReason(reasons, "Low disk space", -5.0, false, offer.DiskSpace)
|
|
} else if offer.DiskSpace >= 750.0 {
|
|
reasons = AddReason(reasons, "Reasonable disk space", 1.0, false, offer.DiskSpace)
|
|
} else if offer.DiskSpace >= 250.0 {
|
|
reasons = AddReason(reasons, "Concerning disk space", -1.0, false, offer.DiskSpace)
|
|
} else if offer.DiskSpace >= 2500.0 {
|
|
reasons = AddReason(reasons, "High disk space", 2.0, false, offer.DiskSpace)
|
|
}
|
|
|
|
// Judge GPU architecture
|
|
if offer.GPUArch == "nvidia" {
|
|
reasons = AddReason(reasons, "Nvidia Preference", 1.0, false, offer.GPUArch)
|
|
} else {
|
|
reasons = AddReason(reasons, "Unknown/Incompatible GPU Architecture", -10.0, false, offer.GPUArch)
|
|
}
|
|
|
|
// Judge reliability
|
|
if offer.Reliability2 < 0.98 {
|
|
reasons = AddReason(reasons, "Low reliability", -5.0, false, offer.Reliability2)
|
|
} else if offer.Reliability2 >= 0.999 {
|
|
reasons = AddReason(reasons, "Very high reliability", 5.0, false, offer.Reliability2)
|
|
} else if offer.Reliability2 >= 0.995 {
|
|
reasons = AddReason(reasons, "High reliability", 2.0, false, offer.Reliability2)
|
|
} else if offer.Reliability2 >= 0.99 {
|
|
reasons = AddReason(reasons, "Decent reliability", 1.0, false, offer.Reliability2)
|
|
}
|
|
|
|
// Calculate base score
|
|
score := 0.0
|
|
for _, reason := range reasons {
|
|
if !reason.IsMultiplier {
|
|
score += reason.Offset
|
|
}
|
|
}
|
|
|
|
// Apply multipliers
|
|
multiplier := 1.0
|
|
for _, reason := range reasons {
|
|
if reason.IsMultiplier {
|
|
multiplier *= reason.Offset
|
|
}
|
|
}
|
|
newScore := score * multiplier
|
|
// sugar.Infow("Multiplier Applied", "offer", offer.ID, "baseScore", score, "score", newScore, "multiplier", multiplier)
|
|
score = newScore
|
|
|
|
scoredOffers = append(scoredOffers, ScoredOffer{Offer: offer, Score: score, Reasons: reasons})
|
|
}
|
|
return scoredOffers
|
|
}
|