• Daniel Hiltgen's avatar
    Support forced spreading for multi GPU · 5e8ff556
    Daniel Hiltgen authored
    Our default behavior today is to try to fit into a single GPU if possible.
    Some users would prefer the old behavior of always spreading across
    multiple GPUs even if the model can fit into one.  This exposes that
    tunable behavior.
    5e8ff556
sched.go 21 KB