mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-12 01:10:22 +00:00
88d28e07a7
The way to run nccl-test over mscclpp: mpirun -np 8 --bind-to numa --allow-run-as-root -x LD_PRELOAD=$(pwd)/build/apps/nccl/libmscclpp_nccl.so -x NCCL_DEBUG=WARN -x MSCCLPP_EXECUTION_PLAN_DIR=/execution-files /root/nccl-tests/build/all_reduce_perf -b 1K -e 1G -f 2 -d half -G 20 -w 10 -n 20
189 lines
4.2 KiB
JSON
189 lines
4.2 KiB
JSON
{
|
|
"name": "send_recv",
|
|
"collective": "sendrecv",
|
|
"protocol": "Simple",
|
|
"inplace": false,
|
|
"gpus": [
|
|
{
|
|
"id": 0,
|
|
"inputChunks": 1,
|
|
"outputChunks": 1,
|
|
"scratchChunks": 2,
|
|
"chunkGroups": 1,
|
|
"threadblocks": [
|
|
{
|
|
"id": 0,
|
|
"ops": [
|
|
{
|
|
"name": "pwsf",
|
|
"o_buff": {
|
|
"src": "i",
|
|
"dst": "s"
|
|
},
|
|
"o_cids": [
|
|
{
|
|
"id": 0,
|
|
"off": 1
|
|
}
|
|
],
|
|
"srcs": [
|
|
{
|
|
"buff": "i",
|
|
"off": 0
|
|
}
|
|
],
|
|
"ctype": "proxy",
|
|
"cnt": 1
|
|
},
|
|
{
|
|
"name": "wait",
|
|
"i_buff": {
|
|
"src": "i",
|
|
"dst": "s"
|
|
},
|
|
"i_cids": [
|
|
{
|
|
"id": 0,
|
|
"off": 0
|
|
}
|
|
],
|
|
"ctype": "proxy",
|
|
"cnt": 1
|
|
},
|
|
{
|
|
"name": "nop",
|
|
"deps": [
|
|
{
|
|
"tb": 0,
|
|
"step": 1
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "copy",
|
|
"src": 0,
|
|
"srcbuff": "s",
|
|
"srcoff": 1,
|
|
"dst": 0,
|
|
"dstbuff": "o",
|
|
"dstoff": 0,
|
|
"ctype": "none",
|
|
"cnt": 1
|
|
}
|
|
],
|
|
"channels": [
|
|
{
|
|
"src": "i",
|
|
"dst": "s",
|
|
"ctype": "proxy",
|
|
"cids": [
|
|
0
|
|
]
|
|
}
|
|
]
|
|
}
|
|
],
|
|
"channels": [
|
|
{
|
|
"srcbuff": "i",
|
|
"dstbuff": "s",
|
|
"type": "proxy",
|
|
"connectedTo": [
|
|
1
|
|
]
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"id": 1,
|
|
"inputChunks": 1,
|
|
"outputChunks": 1,
|
|
"scratchChunks": 2,
|
|
"chunkGroups": 1,
|
|
"threadblocks": [
|
|
{
|
|
"id": 0,
|
|
"ops": [
|
|
{
|
|
"name": "pwsf",
|
|
"o_buff": {
|
|
"src": "i",
|
|
"dst": "s"
|
|
},
|
|
"o_cids": [
|
|
{
|
|
"id": 0,
|
|
"off": 1
|
|
}
|
|
],
|
|
"srcs": [
|
|
{
|
|
"buff": "i",
|
|
"off": 0
|
|
}
|
|
],
|
|
"ctype": "proxy",
|
|
"cnt": 1
|
|
},
|
|
{
|
|
"name": "wait",
|
|
"i_buff": {
|
|
"src": "i",
|
|
"dst": "s"
|
|
},
|
|
"i_cids": [
|
|
{
|
|
"id": 0,
|
|
"off": 0
|
|
}
|
|
],
|
|
"ctype": "proxy",
|
|
"cnt": 1
|
|
},
|
|
{
|
|
"name": "nop",
|
|
"deps": [
|
|
{
|
|
"tb": 0,
|
|
"step": 1
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "copy",
|
|
"src": 1,
|
|
"srcbuff": "s",
|
|
"srcoff": 1,
|
|
"dst": 1,
|
|
"dstbuff": "o",
|
|
"dstoff": 0,
|
|
"ctype": "none",
|
|
"cnt": 1
|
|
}
|
|
],
|
|
"channels": [
|
|
{
|
|
"src": "i",
|
|
"dst": "s",
|
|
"ctype": "proxy",
|
|
"cids": [
|
|
0
|
|
]
|
|
}
|
|
]
|
|
}
|
|
],
|
|
"channels": [
|
|
{
|
|
"srcbuff": "i",
|
|
"dstbuff": "s",
|
|
"type": "proxy",
|
|
"connectedTo": [
|
|
0
|
|
]
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}
|