Skip to content

Commit dcfef94

Browse files
committed
Added large scale 2k parameters sample shift
1 parent 1c5552b commit dcfef94

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

configs/spmd/large_scale/large_scale.json

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,15 @@
2727
"mpi_params": {"n": [1,2,6,12,24,48,96,192,384,768,1536,3072,6144,12288,24576], "ppn": 12, "-hostfile": "", "-cpu-bind=list:0-7,104-111:8-15,112-119:16-23,120-127:24-31,128-135:32-39,136-143:40-47,144-151:52-59,156-163:60-67,164-171:68-75,172-179:76-83,180-187:84-91,188-195:92-99,196-203": "--envall gpu_tile_compact.sh" }
2828
}
2929
},
30+
"large scale 2k parameters sample shift": {
31+
"data": {
32+
"dtype": "float64",
33+
"distributed_split": "sample_shift"
34+
},
35+
"bench": {
36+
"mpi_params": {"n": [1,2,6,12], "ppn": 12, "-hostfile": "", "-cpu-bind=list:0-7,104-111:8-15,112-119:16-23,120-127:24-31,128-135:32-39,136-143:40-47,144-151:52-59,156-163:60-67,164-171:68-75,172-179:76-83,180-187:84-91,188-195:92-99,196-203": "--envall gpu_tile_compact.sh" }
37+
}
38+
},
3039
"large scale 32 parameters": {
3140
"data": {
3241
"dtype": "float64",

0 commit comments

Comments
 (0)