We provide the artifact for the ASPLOS 2023 paper "ElasticFlow: An Elastic Serverless Training Platform for Distributed Deep Learning", including:
- The main implementation of ElasticFlow.
- Cluster simulation scripts (Sec 6.3 & 6.4 & 6.5), which get the main results of the paper.
- Testbed experiment scripts (Sec 6.2 & 6.6).
- Figure plotting scripts.
The simulation experiments need jobs traces that can not make public for now due to policy reasons. We are trying our best to make the trace files public as soon as possible. To get the traces, you need to sign an NDA (non-disclosure agreement) with our company. Please send us your name, affiliation, and address so that we can sign NDA with you and send you the job traces privately.
The descriptions of the job traces are in private_data.md.
Please see ElasticFlow/README.md for more details.
Please see pollux/pollux_simulator/README.md for more details.
Note: Due to the execution scripts of testbed experiments are highly related to internal testbed platform, we only demonstrate the functionality and provide the reproduction steps on the hardware devices we use. Please adjust to your platform if you would like to execute the testbed experiment.
The testbed experiments require 16 nodes, each with 8 A100 GPUs, 96 CPU cores, 900 GB RAM, and eight NVIDIA Mellanox HDR InfiniBand HCAs. You may use the Azure Standard_ND96asr_A100 VMs for reproduction.
The testbed experiments need jobs traces that can not make public for now due to policy reasons. We are trying our best to make the trace files public as soon as possible. To get the traces, you need to sign an NDA (non-disclosure agreement) with our company. Please send us your name, affiliation, and address so that we can sign NDA with you and send you the job traces.
The descriptions of the job traces are in private_data.md.
Please see ElasticFlow/README.md for more details.
As the Pollux baseline is implemented on k8s, we do not interage Pollux in the ElasticFlow system for comparison. We use the open-sourced artifact from the Pollux repo for testbed experiments.
Please see pollux/pollux_testbed/README.md for more details.
Please refer to <repo>/plot_figure/README.md