Skip to content

Commit

Permalink
adding sparse attention to feature index page (microsoft#377)
Browse files Browse the repository at this point in the history
  • Loading branch information
arashashari authored Sep 9, 2020
1 parent 9dadf38 commit b73894d
Showing 1 changed file with 19 additions and 0 deletions.
19 changes: 19 additions & 0 deletions docs/_pages/features.md
Original file line number Diff line number Diff line change
Expand Up @@ -221,3 +221,22 @@ Please see the [core API doc](https://deepspeed.readthedocs.io/) for more detail
}
}
```
## Sparse Attention
DeepSpeed offers sparse attention to support long sequences. Please refer to the [Sparse Attention](/tutorials/sparse_attention/) tutorial.

```python
--deepspeed_sparse_attention
```

```python
"sparse_attention": {
"mode": "fixed",
"block": 16,
"different_layout_per_head": true,
"num_local_blocks": 4,
"num_global_blocks": 1,
"attention": "bidirectional",
"horizontal_global_attention": false,
"num_different_global_patterns": 4
}
```

0 comments on commit b73894d

Please sign in to comment.