@@ -59,9 +59,10 @@ const (
59
59
devfsPath = "dev"
60
60
mib = 1024.0 * 1024.0
61
61
// null device major, minor on linux.
62
- devNullMajor = 1
63
- devNullMinor = 3
64
- devNullType = unix .S_IFCHR
62
+ devNullMajor = 1
63
+ devNullMinor = 3
64
+ devNullType = unix .S_IFCHR
65
+ // GPU connectivity
65
66
maxK8sLabelSize = 63
66
67
fullyConnected = "FULL"
67
68
)
@@ -299,29 +300,20 @@ func generateDriFiles(opts genOptions) {
299
300
func makeXelinkSideCar (topology string , gpus , tiles int , connections string ) {
300
301
if topology != fullyConnected {
301
302
log .Printf ("XELINK: generate xelink sidecar label file, using (GPUs: %d, Tiles: %d)" , gpus , tiles )
303
+ saveSideCarFile (connections )
302
304
} else {
303
305
log .Printf ("XELINK: generate xelink sidecar label file, using (GPUs: %d, Tiles: %d, Topology: %s)" , gpus , tiles , topology )
304
- }
305
-
306
- if topology == fullyConnected {
307
306
saveSideCarFile (buildConnectionList (gpus , tiles ))
308
- } else {
309
- saveSideCarFile (connections )
310
307
}
311
308
}
312
309
313
310
func buildConnectionList (gpus , tiles int ) string {
314
- mm := 0
315
-
316
311
var nodes = make ([]string , 0 )
317
312
318
- for mm < gpus {
319
- nn := 0
320
- for nn < tiles {
313
+ for mm := 0 ; mm < gpus ; mm ++ {
314
+ for nn := 0 ; nn < tiles ; nn ++ {
321
315
nodes = append (nodes , fmt .Sprintf ("%d.%d" , mm , nn ))
322
- nn ++
323
316
}
324
- mm ++
325
317
}
326
318
327
319
var links = make (map [string ]struct {}, 0 )
0 commit comments