You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
] # this ensures that people won't have to clone the whole repo to include notebooks, they can just do `pip install kernel_tuner[tutorial,cuda]`
49
+
50
+
# ATTENTION: if anything is changed here, run `poetry update`
51
+
requires-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
52
+
dependencies = [
53
+
"numpy (>=1.26.0,<2.0.0)", # Python 3.12 requires numpy at least 1.26, CuPy does not support 2.0
54
+
"scipy>=1.14.1",
55
+
"packaging", # required by file_utils
56
+
"jsonschema",
57
+
"python-constraint2>=2.2.2",
58
+
"xmltodict",
59
+
"pandas>=2.0.0",
60
+
"scikit-learn>=1.0.2",
61
+
]
62
+
# NOTE Torch can be used with Kernel Tuner, but is not a dependency, should be up to the user to use it
# ATTENTION: if anything is changed here, run `poetry update`
66
-
[tool.poetry.dependencies]
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
pyopencl = { version = "*", optional = true } # Attention: if pyopencl is changed here, also change `session.install("pyopencl")` in the Noxfile
90
-
# HIP
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
92
-
# Tutorial (for the notebooks used in the examples)
93
-
jupyter = { version = "^1.0.0", optional = true }
94
-
matplotlib = { version = "^3.5.0", optional = true }
95
86
96
-
[tool.poetry.extras]
97
-
cuda = ["pycuda", "nvidia-ml-py", "pynvml"]
98
-
opencl = ["pyopencl"]
99
-
cuda_opencl = ["pycuda", "pyopencl"]
87
+
[[tool.poetry.source]]
88
+
name = "testpypi"
89
+
url = "https://test.pypi.org/simple/"
90
+
priority = "explicit"
91
+
92
+
[tool.poetry.dependencies]
93
+
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 and move this to [project.optional-dependencies]
94
+
95
+
[project.optional-dependencies]
96
+
cuda = ["pycuda>=2025.1", "nvidia-ml-py>=12.535.108", "pynvml>=11.4.1"] # Attention: if pycuda is changed here, also change `session.install("pycuda")` in the Noxfile
97
+
opencl = ["pyopencl"] # Attention: if pyopencl is changed here, also change `session.install("pyopencl")` in the Noxfile
98
+
cuda_opencl = ["pycuda>=2024.1", "pyopencl"] # Attention: if pycuda is changed here, also change `session.install("pycuda")` in the Noxfile
# ATTENTION: if anything is changed here, run `poetry update` and `poetry export --with docs --without-hashes --format=requirements.txt --output doc/requirements.txt`
104
103
# Please note that there is overlap with the `dev` group
0 commit comments