-
Notifications
You must be signed in to change notification settings - Fork 6
Expand file tree
/
Copy pathpyproject.toml
More file actions
30 lines (25 loc) · 1.09 KB
/
pyproject.toml
File metadata and controls
30 lines (25 loc) · 1.09 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
[build-system]
requires = ["scikit-build-core", "pybind11", "libclang"]
build-backend = "scikit_build_core.build"
[project]
name = "InfiniOps"
version = "0.1.0"
[project.optional-dependencies]
# TODO: `torch` here is unconstrained. On Ascend hosts, the working
# torch is the Ascend-matched `torch 2.9.0+cpu` paired with
# `torch_npu 2.9.0.post1+…`. A `pip install -e .[dev] --force-reinstall`
# will re-resolve `torch` to the latest PyPI version (currently
# `torch 2.11.0`), which now declares `cuda-toolkit` / `nvidia-cublas` /
# `nvidia-cudnn` / … as hard deps — downloads GBs of CUDA wheels and
# kills the `torch_npu` / `vllm-ascend` pairing. Needs a platform-aware
# split (e.g. `torch; platform_machine != 'aarch64'`, or move `torch`
# out of `dev` and require it pre-installed in the container image).
dev = ["pytest", "pytest-cov", "pytest-xdist", "ruff", "torch", "pyyaml"]
[tool.scikit-build.wheel]
install-dir = "infini"
[tool.scikit-build.cmake.define]
AUTO_DETECT_DEVICES = "ON"
AUTO_DETECT_BACKENDS = "ON"
GENERATE_PYTHON_BINDINGS = "ON"
[tool.pytest.ini_options]
testpaths = ["tests"]