Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 13 additions & 7 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api"
name = "kernel_tuner"
packages = [{ include = "kernel_tuner", from = "." }]
description = "An easy to use CUDA/OpenCL kernel tuner in Python"
version = "1.0" # adhere to PEP440 versioning: https://packaging.python.org/en/latest/guides/distributing-packages-using-setuptools/#id55
version = "1.1.0" # adhere to PEP440 versioning: https://packaging.python.org/en/latest/guides/distributing-packages-using-setuptools/#id55
license = "Apache-2.0"
authors = [
"Ben van Werkhoven <b.vanwerkhoven@esciencecenter.nl>",
Expand Down Expand Up @@ -57,12 +57,17 @@ generate-setup-file = false
[tool.poetry.scripts]
kernel_tuner = "kernel_tuner.interface:entry_point"

[[tool.poetry.source]]
name = "testpypi"
url = "https://test.pypi.org/simple/"
priority = "explicit"

# ATTENTION: if anything is changed here, run `poetry update`
[tool.poetry.dependencies]
python = ">=3.9,<3.15" # NOTE when changing the supported Python versions, also change the test versions in the noxfile
numpy = "^1.26.0" # Python 3.12 requires numpy at least 1.26
scipy = ">=1.11.0" # held back by Python 3.9
packaging = "*" # required by file_utils
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
numpy = "^1.26.0" # Python 3.12 requires numpy at least 1.26
scipy = ">=1.11.0" # held back by Python 3.9
packaging = "*" # required by file_utils
jsonschema = "*"
python-constraint2 = "^2.1.0"
xmltodict = "*"
Expand All @@ -72,6 +77,7 @@ scikit-learn = ">=1.0.2"

# List of optional dependencies for user installation, e.g. `pip install kernel_tuner[cuda]`, used in the below `extras`.
# Please note that this is different from the dependency groups below, e.g. `docs` and `test`, those are for development.
# ATTENTION: if anything is changed here, run `poetry update`
# CUDA
pycuda = { version = "^2024.1", optional = true } # Attention: if pycuda is changed here, also change `session.install("pycuda")` in the Noxfile
nvidia-ml-py = { version = "^12.535.108", optional = true }
Expand All @@ -82,7 +88,7 @@ pynvml = { version = "^11.4.1", optional = true }
# OpenCL
pyopencl = { version = "*", optional = true } # Attention: if pyopencl is changed here, also change `session.install("pyopencl")` in the Noxfile
# HIP
hip-python-fork = { version = "*", optional = true }
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
# Tutorial (for the notebooks used in the examples)
jupyter = { version = "^1.0.0", optional = true }
matplotlib = { version = "^3.5.0", optional = true }
Expand All @@ -91,7 +97,7 @@ matplotlib = { version = "^3.5.0", optional = true }
cuda = ["pycuda", "nvidia-ml-py", "pynvml"]
opencl = ["pyopencl"]
cuda_opencl = ["pycuda", "pyopencl"]
hip = ["hip-python-fork"]
hip = ["hip-python"]
tutorial = ["jupyter", "matplotlib", "nvidia-ml-py"]

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