ASTRA-Sim: Quick Example
Run Astra-Sim with a quick example (Ring_allgather_16npus.sh): This example demonstrates a ring all-gather operation on 16 NPUs connected in a ring topology. Please go back to build up the necessary dependencies before running the example. The following shows the expected output:
1[system::topology::RingTopology] [info] ring of node 0, id: 0 dimension: local total nodes in ring: 16 index in ring: 0 offset: 1 total nodes in ring: 16
2[system::topology::RingTopology] [info] ring of node 0, id: 0 dimension: local total nodes in ring: 16 index in ring: 0 offset: 1 total nodes in ring: 16
3[system::topology::RingTopology] [info] ring of node 0, id: 0 dimension: local total nodes in ring: 16 index in ring: 0 offset: 1 total nodes in ring: 16
4[system::topology::RingTopology] [info] ring of node 0, id: 0 dimension: local total nodes in ring: 16 index in ring: 0 offset: 1 total nodes in ring: 16
5[workload] [info] sys[12] finished, 323560 cycles, exposed communication 323560 cycles.
6[workload] [info] sys[13] finished, 323560 cycles, exposed communication 323560 cycles.
7[workload] [info] sys[14] finished, 323560 cycles, exposed communication 323560 cycles.
8[workload] [info] sys[15] finished, 323560 cycles, exposed communication 323560 cycles.
9[workload] [info] sys[0] finished, 323560 cycles, exposed communication 323560 cycles.
10[workload] [info] sys[1] finished, 323560 cycles, exposed communication 323560 cycles.
11[workload] [info] sys[2] finished, 323560 cycles, exposed communication 323560 cycles.
12[workload] [info] sys[3] finished, 323560 cycles, exposed communication 323560 cycles.
13[workload] [info] sys[4] finished, 323560 cycles, exposed communication 323560 cycles.
14[workload] [info] sys[5] finished, 323560 cycles, exposed communication 323560 cycles.
15[workload] [info] sys[6] finished, 323560 cycles, exposed communication 323560 cycles.
16[workload] [info] sys[7] finished, 323560 cycles, exposed communication 323560 cycles.
17[workload] [info] sys[8] finished, 323560 cycles, exposed communication 323560 cycles.
18[workload] [info] sys[9] finished, 323560 cycles, exposed communication 323560 cycles.
19[workload] [info] sys[10] finished, 323560 cycles, exposed communication 323560 cycles.
20[workload] [info] sys[11] finished, 323560 cycles, exposed communication 323560 cycles.
For additional quick runs and other featured microbenchmarks, see the ASTRA-Sim Examples and review the accompanying documentation.