Skip to content

Commit 3b9f5df

Browse files
committed
Updated pyproject to use more recent hip-python from testpypi
1 parent c1acc3b commit 3b9f5df

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

pyproject.toml

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -57,9 +57,14 @@ generate-setup-file = false
5757
[tool.poetry.scripts]
5858
kernel_tuner = "kernel_tuner.interface:entry_point"
5959

60+
[[tool.poetry.source]]
61+
name = "testpypi"
62+
url = "https://test.pypi.org/simple/"
63+
priority = "explicit"
64+
6065
# ATTENTION: if anything is changed here, run `poetry update`
6166
[tool.poetry.dependencies]
62-
python = ">=3.10,<4" # <4 is because of hip-python-fork # NOTE when changing the Python versions, also change the test versions in the Noxfile and GitHub Actions
67+
python = ">=3.10,<4" # <4 is because of hip-python # NOTE when changing the Python versions, also change the test versions in the Noxfile and GitHub Actions
6368
numpy = "^1.26.0" # Python >3.12 requires numpy >= 1.26
6469
scipy = ">=1.14.1"
6570
packaging = "*" # required by file_utils
@@ -72,6 +77,7 @@ scikit-learn = ">=1.0.2"
7277

7378
# List of optional dependencies for user installation, e.g. `pip install kernel_tuner[cuda]`, used in the below `extras`.
7479
# Please note that this is different from the dependency groups below, e.g. `docs` and `test`, those are for development.
80+
# ATTENTION: if anything is changed here, run `poetry update`
7581
# CUDA
7682
pycuda = { version = "^2025.1", optional = true } # Attention: if pycuda is changed here, also change `session.install("pycuda")` in the Noxfile
7783
nvidia-ml-py = { version = "^12.535.108", optional = true }
@@ -82,7 +88,7 @@ pynvml = { version = "^11.4.1", optional = true }
8288
# OpenCL
8389
pyopencl = { version = "*", optional = true } # Attention: if pyopencl is changed here, also change `session.install("pyopencl")` in the Noxfile
8490
# HIP
85-
hip-python-fork = { version = "*", optional = true }
91+
hip-python = { version = "^6.3.3.540.31", source = "testpypi", optional = true } # Note: when released, switch this package to pypi and remove tool.poetry.source
8692
# Tutorial (for the notebooks used in the examples)
8793
jupyter = { version = "^1.0.0", optional = true }
8894
matplotlib = { version = "^3.5.0", optional = true }
@@ -91,7 +97,7 @@ matplotlib = { version = "^3.5.0", optional = true }
9197
cuda = ["pycuda", "nvidia-ml-py", "pynvml"]
9298
opencl = ["pyopencl"]
9399
cuda_opencl = ["pycuda", "pyopencl"]
94-
hip = ["hip-python-fork"]
100+
hip = ["hip-python"]
95101
tutorial = ["jupyter", "matplotlib", "nvidia-ml-py"]
96102

97103
# ATTENTION: if anything is changed here, run `poetry update` and `poetry export --with docs --without-hashes --format=requirements.txt --output doc/requirements.txt`

0 commit comments

Comments
 (0)