[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", ]