Skip to content

Commit f1631a6

Browse files
committed
Update pyproject.toml tileiras dependency
Signed-off-by: Jay Gu <jagu@nvidia.com>
1 parent ac755a3 commit f1631a6

2 files changed

Lines changed: 16 additions & 10 deletions

File tree

docs/source/quickstart.rst

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -25,10 +25,12 @@ cuTile Python requires the following:
2525
Installing cuTile Python
2626
-------------------------------------------------------------------------------
2727

28-
cuTile Python depends on CUDA TileIR compiler ``tileiras``, which futher depends on ``ptxas`` and ``libnvvm``
29-
from the CUDA Toolkit.
28+
cuTile Python depends on CUDA TileIR compiler ``tileiras``,
29+
which futher depends on ``ptxas`` and ``libnvvm`` from the CUDA Toolkit. They can be installed
30+
as python packages per virtual environment or at system wide CTK locations.
3031

31-
If your system does not have system-wide CUDA Toolkit (13.1+), you can install cuTile Python along with ``[tileiras]``,
32+
If your system does not have system-wide CUDA Toolkit (13.1+),
33+
you can install cuTile Python along with the optional ``[tileiras]``,
3234
which installs ``nvidia-cuda-tileiras``, ``nvidia-cuda-nvcc`` and
3335
``nvidia-nvvm`` into your Python virtual environment.
3436

@@ -39,9 +41,17 @@ which installs ``nvidia-cuda-tileiras``, ``nvidia-cuda-nvcc`` and
3941
Note: the package versions for ``nvidia-cuda-tileiras``, ``nvidia-cuda-nvcc`` and
4042
``nvidia-nvvm`` must match up to the same major.minor version.
4143

44+
cuTile python supports different versions of `tileiras`.
45+
To use a specific version of ``tileiras`` python pacakge, i.e. `13.3.0` run
4246

43-
Alternatively if you already have system-wide CUDA Toolkit (13.1+) installed, you can install cuTile Python as a
44-
standalone package. cuTile automatically searches for ``tileiras`` from the location of CUDA Toolkit.
47+
.. code-block:: bash
48+
49+
pip install cuda-toolkit[tileiras,nvvm,nvcc]==13.3.0
50+
51+
52+
Alternatively if you already have system-wide CUDA Toolkit (13.1+) installed,
53+
you can install cuTile Python as a standalone package.
54+
cuTile automatically searches for ``tileiras`` from the location of CUDA Toolkit.
4555

4656
.. code-block:: bash
4757

pyproject.toml

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,7 @@ dependencies = [
3939
]
4040

4141
[project.optional-dependencies]
42-
tileiras = [
43-
"nvidia-cuda-tileiras>=13.2,<13.3",
44-
"nvidia-cuda-nvcc>=13.2,<13.3",
45-
"nvidia-nvvm>=13.2,<13.3",
46-
]
42+
tileiras = [ "cuda-toolkit[tileiras,nvcc,nvvm]>=13.2,<13.4" ]
4743

4844
[project.urls]
4945
Homepage = "https://github.com/nvidia/cutile-python"

0 commit comments

Comments
 (0)