fardeen786/astra-sim repository overview

⁠ASTRA-sim — prebuilt

ASTRA-sim compiled and ready to run, with the Chakra execution-trace tools and the STAGE workload generator. No toolchain to build, no submodules to clone, no protobuf version conflicts.

Tags

⁠Quick start

docker pull fardeen786/astra-sim:latest

docker run --rm -it \
  -v "$PWD:/workspace" \
  --user "$(id -u):$(id -g)" \
  fardeen786/astra-sim:latest

--user keeps output files owned by you instead of root. Your current directory appears at /workspace.

Handy alias:

alias astra='docker run --rm -it -v "$PWD:/workspace" --user "$(id -u):$(id -g)" fardeen786/astra-sim:latest'

⁠Commands

Example configs are at /opt/astra-sim/examples:

E=/opt/astra-sim/examples
astra-sim-analytical \
  --workload-configuration=$E/workload/microbenchmarks/all_reduce/8npus_1MB/all_reduce \
  --system-configuration=$E/system/native_collectives/HGX-H100-validated.json \
  --network-configuration=$E/network/analytical/HGX-H100-validated.yml \
  --remote-memory-configuration=$E/remote_memory/analytical/no_memory_expansion.json

Generate your own workload and simulate it:

stage --output_dir ./gen --output_name 'workload.%d.et' --dp 2 --tp 2 --pp 2
astra-sim-analytical \
  --workload-configuration=./gen/workload \
  --comm-group-configuration=./gen/workload.json \
  --system-configuration=$E/system/native_collectives/Ring_4chunks.json \
  --network-configuration=$E/network/analytical/Ring_8npus.yml \
  --remote-memory-configuration=$E/remote_memory/analytical/no_memory_expansion.json

⁠Notes

• htsim seeding. htsim chooses ECMP paths randomly; this image seeds it from ASTRA_HTSIM_SEED (default 1) so runs reproduce. Upstream seeds from the wall clock, which makes identical runs disagree. The path choice changes the answer by up to ~33%, so report the seed and prefer sweeping several. A small fraction of seeds hit an upstream bug and abort — change the seed if so.
• x86-64 only.
• README.txt inside the container lists the commands on PATH and how to rebuild the binaries. Run cat README.txt in the shell.

⁠Contents

ASTRA-sim 518bd513 · STAGE 71ceb423 · PARAM 5ee3dc2f · ns-3 3.42 (MPI) · Protobuf 29.0 / 5.29.0

Upstream project: https://github.com/astra-sim/astra-sim⁠ · Docs: https://astra-sim.github.io/astra-sim-docs/⁠