Skip to content

Add AutoSP example#999

Draft
neeldani wants to merge 4 commits intodeepspeedai:masterfrom
neeldani:autosp-example
Draft

Add AutoSP example#999
neeldani wants to merge 4 commits intodeepspeedai:masterfrom
neeldani:autosp-example

Conversation

@neeldani
Copy link

@neeldani neeldani commented Feb 23, 2026

This PR adds an example of using AutoSP

Reference: deepspeedai/DeepSpeed#7860

@@ -0,0 +1,52 @@
# AutoSP Setup Guide
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please add a brief description of the files in this folder

```

#### Run autosp on 2 GPUs
```bash
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Include screenshot/snippet of expected output, this is helpful for users.

# AutoSP Setup Guide

Quick start guide to clone and set up the AutoSP repository.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Include a brief description that this example demonstrates usage in HF accelerate.

type=str,
default="autosp",
choices=["eager", "compile", "autosp", "ringattn"],
help="Compilation mode: eager (no compilation), compile (torch.compile), autosp (AutoSP), ringattn (ring attention)"
Copy link
Contributor

@tjruwase tjruwase Mar 1, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

These options should discussed in the README intro as the included examples.

type=str,
default="autosp",
choices=["eager", "compile", "autosp", "ringattn"],
help="Compilation mode: eager (no compilation), compile (torch.compile), autosp (AutoSP), ringattn (ring attention)"
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The example runs listed in the README seems to exclude the ring attention option. Is this intentional?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants