Archived
1
0
Fork 0

feat: Add NVIDIA FE (#15)

Signed-off-by: Julien Riou <julien@riou.xyz>
This commit is contained in:
Julien Riou 2022-07-22 19:34:03 +02:00
parent 999018492e
commit 991880f1c9
No known key found for this signature in database
GPG key ID: FF42D23B580C89F7
4 changed files with 228 additions and 0 deletions

14
main.go
View file

@ -216,6 +216,20 @@ func main() {
}
}
if config.HasNvidiaFE() {
// create a parser for all locations
for _, location := range config.NvidiaFEConfig.Locations {
parser, err := NewNvidiaFRParser(location, config.NvidiaFEConfig.GPUs, config.NvidiaFEConfig.UserAgent, config.NvidiaFEConfig.Timeout)
if err != nil {
log.Warnf("could not create NVIDIA FE parser for location %s: %s", location, err)
continue
}
parsers = append(parsers, parser)
log.Debugf("parser %s registered", parser)
}
}
// parse asynchronously
var wg sync.WaitGroup
jobsCount := 0