Skip to content

Commit 0de3d84

Browse files
committed
Updated pyproject to use more recent hip-python from testpypi
1 parent 28014d7 commit 0de3d84

File tree

1 file changed

+13
-7
lines changed

1 file changed

+13
-7
lines changed

pyproject.toml

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api"
66
name = "kernel_tuner"
77
packages = [{ include = "kernel_tuner", from = "." }]
88
description = "An easy to use CUDA/OpenCL kernel tuner in Python"
9-
version = "1.0" # adhere to PEP440 versioning: https://packaging.python.org/en/latest/guides/distributing-packages-using-setuptools/#id55
9+
version = "1.1.0" # adhere to PEP440 versioning: https://packaging.python.org/en/latest/guides/distributing-packages-using-setuptools/#id55
1010
license = "Apache-2.0"
1111
authors = [
1212
"Ben van Werkhoven <b.vanwerkhoven@esciencecenter.nl>",
@@ -57,12 +57,17 @@ 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.9,<3.15" # NOTE when changing the supported Python versions, also change the test versions in the noxfile
63-
numpy = "^1.26.0" # Python 3.12 requires numpy at least 1.26
64-
scipy = ">=1.11.0" # held back by Python 3.9
65-
packaging = "*" # required by file_utils
67+
python = ">=3.9,<4" # <4 is because of hip-python # NOTE when changing the supported Python versions, also change the test versions in the noxfile
68+
numpy = "^1.26.0" # Python 3.12 requires numpy at least 1.26
69+
scipy = ">=1.11.0" # held back by Python 3.9
70+
packaging = "*" # required by file_utils
6671
jsonschema = "*"
6772
python-constraint2 = "^2.1.0"
6873
xmltodict = "*"
@@ -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 = "^2024.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)