Skip to content

Repository files navigation

StreamEP-Artifact

For SOSP'26 artifact review, this repo (including submodules) contains

  • StreamInfer: prototype distributed MoE decoding system implementing StreamEP. Currently only work with dummy model weights, yet it performs all model computations and data movements, and can replay authentic expert routing profiled from real model execution of each dataset.
  • sglang_dummy_prefill: a fork of sglang with modifications to support decoding from dummy KV-Cache tensors. It can also replay expert routing from profiles. It also supports other minor options (e.g. skipping the 1st dense layer in GLM) for fair comparison with StreamInfer.
  • experiment_utils: Instructions and scripts for artifact reviewers.

Getting started

On the provided testbed (SPHERE) everything is already installed — go straight to experiment_utils/ and run the two experiments below.

Note that the first main experiment in the paper draft (Figure 10) is about the peak throughput, and the throughput-itl experiment (Figure 11) is an extention of it, so we directly start with figure 11 for reproduction. To limit the total combinations of experiments, we only run GPT-OSS-120b on the SPHERE testbed, as the Delta testbed is not reservable. The next experiment we reproduce is the tolerance to network interference (Figure 13).

Each of the above have two scripts, one for StreamInfer and one for baseline, so there are 4 major experiment scripts to run. On sphere, each script is about 30 minutes, so about 2 hours are needed in total. Note that the SPHERE cluster has one node's network that is sometimes flaky (distributed engine boots can hang on it), which is why the run scripts retry failed boots/benchmarks automatically — occasional retry messages in the output are expected.

On any other cluster, install both systems on every node first:

About

No description, website, or topics provided.

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages