46 lines
1.0 KiB
TOML
46 lines
1.0 KiB
TOML
[build-system]
|
|
requires = ["setuptools>=68", "wheel"]
|
|
build-backend = "setuptools.build_meta"
|
|
|
|
[project]
|
|
name = "kernel-lab"
|
|
version = "0.1.0"
|
|
description = "Educational GPU kernel lab for PyTorch, Triton, and CUDA."
|
|
readme = "README.md"
|
|
requires-python = ">=3.10"
|
|
dependencies = [
|
|
"numpy>=1.26",
|
|
"torch>=2.10",
|
|
"triton>=3.0",
|
|
"pytest>=8.0",
|
|
"packaging>=24.0",
|
|
"cmake>=3.25",
|
|
"ninja>=1.11",
|
|
]
|
|
|
|
[tool.uv.sources]
|
|
torch = { index = "pytorch-cu128" }
|
|
|
|
[[tool.uv.index]]
|
|
name = "aliyun-pypi"
|
|
url = "https://mirrors.aliyun.com/pypi/simple/"
|
|
default = true
|
|
|
|
[[tool.uv.index]]
|
|
name = "pytorch-cu128"
|
|
url = "https://download.pytorch.org/whl/cu128"
|
|
explicit = true
|
|
|
|
[tool.setuptools.packages.find]
|
|
include = ["reference*", "kernels*"]
|
|
|
|
[tool.pytest.ini_options]
|
|
addopts = "-ra --import-mode=importlib"
|
|
testpaths = ["tests", "tasks"]
|
|
markers = [
|
|
"reference: tests for plain PyTorch references",
|
|
"skeleton: tests validating learner skeleton behavior",
|
|
"cuda_required: tests that need CUDA",
|
|
"triton_required: tests that need Triton",
|
|
]
|