Initial project scaffold
This commit is contained in:
45
pyproject.toml
Normal file
45
pyproject.toml
Normal 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",
|
||||
]
|
||||
Reference in New Issue
Block a user