Skip to content

Commit e1ff9a0

Browse files
experiment with ppn
1 parent fce0651 commit e1ff9a0

File tree

2 files changed

+48
-0
lines changed

2 files changed

+48
-0
lines changed

configs/spmd/large_scale/large_scale.json

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,24 @@
1818
"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" }
1919
}
2020
},
21+
"large scale one node parameters": {
22+
"data": {
23+
"dtype": "float64",
24+
"distributed_split": "None"
25+
},
26+
"bench": {
27+
"mpi_params": {"n": [1,2,6,12], "ppn": [1,2,6,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" }
28+
}
29+
},
30+
"large scale strong one node parameters": {
31+
"data": {
32+
"dtype": "float64",
33+
"distributed_split": "rank_based"
34+
},
35+
"bench": {
36+
"mpi_params": {"n": [1,2,6,12], "ppn": [1,2,6,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+
},
2139
"large scale 2k parameters": {
2240
"data": {
2341
"dtype": "float64",
Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
{
2+
"INCLUDE": ["../../common/sklearn.json", "../../regular/pca.json", "large_scale.json"],
3+
"PARAMETERS_SETS": {
4+
"spmd pca parameters": {
5+
"algorithm": {
6+
"estimator": "PCA",
7+
"estimator_methods": { "training": "fit", "inference": "" }
8+
},
9+
"data": {
10+
"split_kwargs": { "test_size": 0.0001 }
11+
}
12+
},
13+
"synthetic data": {
14+
"data": [
15+
{ "source": "make_blobs", "generation_kwargs": { "n_samples": 10000000, "n_features": 10, "centers": 1 } },
16+
{ "source": "make_blobs", "generation_kwargs": { "n_samples": 100000, "n_features": 1000, "centers": 1 } }
17+
]
18+
}
19+
},
20+
"TEMPLATES": {
21+
"linreg": {
22+
"SETS": [
23+
"sklearnex spmd implementation",
24+
"large scale one node parameters",
25+
"synthetic data",
26+
"spmd pca parameters"
27+
]
28+
}
29+
}
30+
}

0 commit comments

Comments
 (0)