Skip to content

Add AutoSP example#999

Open
neeldani wants to merge 13 commits intodeepspeedai:masterfrom
neeldani:autosp-example
Open

Add AutoSP example#999
neeldani wants to merge 13 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

Copy link
Author

Choose a reason for hiding this comment

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

Done, added here

```

#### 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.

Copy link
Author

Choose a reason for hiding this comment

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

Makes sense, added it here

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?

Copy link
Author

Choose a reason for hiding this comment

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

Thank you for catching this, just updated it

@neeldani neeldani marked this pull request as ready for review March 12, 2026 06:32
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