|
1 | 1 | <p align="center"> |
2 | | - <img src="docs/images/logo.svg" alt="tensor-layouts" width="520"> |
| 2 | + <img src="https://raw.githubusercontent.com/facebookresearch/tensor-layouts/main/docs/images/logo.svg" alt="tensor-layouts" width="520"> |
3 | 3 | </p> |
4 | 4 |
|
5 | 5 | [](https://github.com/facebookresearch/tensor-layouts/actions/workflows/ci.yml) |
6 | 6 | [](https://pypi.org/project/tensor-layouts/) |
7 | | -[](LICENSE) |
| 7 | +[](https://github.com/facebookresearch/tensor-layouts/blob/main/LICENSE) |
8 | 8 |
|
9 | 9 | A pure-Python implementation of the [NVIDIA CuTe](https://github.com/NVIDIA/cutlass/blob/main/media/docs/cute/00_quickstart.md) layout algebra. **No GPU required.** |
10 | 10 |
|
@@ -116,23 +116,23 @@ draw_swizzle(Layout((8, 8), (8, 1)), Swizzle(3, 0, 3), colorize=True) |
116 | 116 | ``` |
117 | 117 |
|
118 | 118 | <p align="center"> |
119 | | - <img src="docs/images/row_major_8x8.png" alt="Row-Major 8x8 layout" width="400"> |
| 119 | + <img src="https://raw.githubusercontent.com/facebookresearch/tensor-layouts/main/docs/images/row_major_8x8.png" alt="Row-Major 8x8 layout" width="400"> |
120 | 120 | </p> |
121 | 121 |
|
122 | 122 | <p align="center"> |
123 | | - <img src="docs/images/swizzle_8x8.png" alt="Swizzle(3, 0, 3) applied to row-major 8x8" width="800"> |
| 123 | + <img src="https://raw.githubusercontent.com/facebookresearch/tensor-layouts/main/docs/images/swizzle_8x8.png" alt="Swizzle(3, 0, 3) applied to row-major 8x8" width="800"> |
124 | 124 | </p> |
125 | 125 |
|
126 | | -See [`examples/viz.ipynb`](examples/viz.ipynb) for a full |
| 126 | +See [`examples/viz.ipynb`](https://github.com/facebookresearch/tensor-layouts/blob/main/examples/viz.ipynb) for a full |
127 | 127 | gallery of layout, swizzle, MMA atom, and tiled MMA visualizations. |
128 | 128 |
|
129 | 129 | ## Documentation |
130 | 130 |
|
131 | | -- [Layout Algebra API](docs/layout_api.md) — construction, querying, compose, complement, divide, product |
132 | | -- [Visualization API](docs/viz_api.md) — draw_layout, draw_swizzle, draw_mma_layout, and more |
133 | | -- [Layout Examples](examples/layouts.py) — runnable script covering the full algebra (run from source checkout with `PYTHONPATH=src python3 examples/layouts.py`) |
134 | | -- [Visualization Examples](examples/viz.py) — runnable script generating all visualization types (run from source checkout with `PYTHONPATH=src python3 examples/viz.py`) |
135 | | -- [Visualization Notebook](examples/viz.ipynb) — Jupyter gallery |
| 131 | +- [Layout Algebra API](https://github.com/facebookresearch/tensor-layouts/blob/main/docs/layout_api.md) — construction, querying, compose, complement, divide, product |
| 132 | +- [Visualization API](https://github.com/facebookresearch/tensor-layouts/blob/main/docs/viz_api.md) — draw_layout, draw_swizzle, draw_mma_layout, and more |
| 133 | +- [Layout Examples](https://github.com/facebookresearch/tensor-layouts/blob/main/examples/layouts.py) — runnable script covering the full algebra (run from source checkout with `PYTHONPATH=src python3 examples/layouts.py`) |
| 134 | +- [Visualization Examples](https://github.com/facebookresearch/tensor-layouts/blob/main/examples/viz.py) — runnable script generating all visualization types (run from source checkout with `PYTHONPATH=src python3 examples/viz.py`) |
| 135 | +- [Visualization Notebook](https://github.com/facebookresearch/tensor-layouts/blob/main/examples/viz.ipynb) — Jupyter gallery |
136 | 136 |
|
137 | 137 | ## Testing |
138 | 138 |
|
@@ -164,4 +164,4 @@ pytest tests/oracle_amd.py |
164 | 164 |
|
165 | 165 | ## License |
166 | 166 |
|
167 | | -MIT License. See [LICENSE](LICENSE) for details. |
| 167 | +MIT License. See [LICENSE](https://github.com/facebookresearch/tensor-layouts/blob/main/LICENSE) for details. |
0 commit comments