Initial project scaffold

This commit is contained in:
2026-04-10 13:22:19 +00:00
commit 7fa69b1354
94 changed files with 3964 additions and 0 deletions

45
pyproject.toml Normal file
View File

@@ -0,0 +1,45 @@
[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",
]