leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

v4.5 tag update (#3202)

Browse Source 
* Python DSL examples reorganization.

* v4.5 tag update.
This commit is contained in:
Junkai-Wu
2026-05-06 08:55:27 +08:00
committed by GitHub
parent f74fea9ce3
commit cb37157db5
351 changed files with 36688 additions and 8117 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
media/docs/pythonDSL/quick_start.rst
View File

@@ -3,19 +3,19 @@
Quick Start Guide
=======================
The CUTLASS DSL 4.4 release currently supports **Linux** and **Python 3.10 - 3.14** only. To install CUTLASS DSLs (limited to CuTe DSL for now), use the following command
Compatibility Requirements
---------------------------------
The CUTLASS DSL 4.4 release currently supports **Linux** and **Python 3.10 - 3.14** only.
Only Linux x86_64 and aarch64 are supported. Additional platform support will be added in future releases.
CUTLASS DSL supports the same NVIDIA driver version as the corresponding `CUDA Toolkit <https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html>`_
(CUDA Toolkit 12.9 or CUDA Toolkit 13.1). Specifically, for 12.9, the driver version must be 575.51.03 or later.
Installation
-----------------------
Before installing the latest version, you need to uninstall any previous CUTLASS DSL Installation.
.. code-block:: bash
pip uninstall nvidia-cutlass-dsl nvidia-cutlass-dsl-libs-base nvidia-cutlass-dsl-libs-cu13 -y
To ensure compatibility with the examples and code on `GitHub <https://github.com/NVIDIA/cutlass/tree/main>`_,
use the `setup.sh <https://github.com/NVIDIA/cutlass/blob/main/python/CuTeDSL/setup.sh>`_ file from the corresponding commit in the repository.
@@ -38,12 +38,10 @@ If you just want to try out the last known stable release of the CUTLASS DSL (ma
pip install nvidia-cutlass-dsl
# For CUDA Toolkit 13.1:
pip install nvidia-cutlass-dsl[cu13]
pip install "nvidia-cutlass-dsl[cu13]"
The ``nvidia-cutlass-dsl`` wheel includes everything needed to generate GPU kernels. It requires
the same NVIDIA driver version as the corresponding `CUDA Toolkit <https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html>`_
(CUDA Toolkit 12.9 or CUDA Toolkit 13.1).
The ``nvidia-cutlass-dsl`` wheel includes everything needed to generate GPU kernels.
Recommended Dependencies
---------------------------------
@@ -52,9 +50,7 @@ To run examples and begin development, we recommend installing:
.. code-block:: bash
pip install torch jupyter
We recommend installing JAX with CUDA support at version 0.8.1 to run JAX examples.
pip install torch jupyter mypy==1.19.1
Recommended Python environment variables for jupyter notebooks
--------------------------------------------------------------