pyproject.toml

# Copyright 2026 Moore Threads
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ==============================================================================


[build-system]
requires = ["setuptools","cmake>=3.18"]
build-backend = "setuptools.build_meta"

[project]
name = "axinfra"
requires-python = ">=3.8" # 3.10 recommended
authors = [{ name = "Moore Threads"}]
dependencies = [
    "numpy==1.26.4",
    "setuptools",
    "psutil",
]

dynamic = ["version", "readme"]

[project.optional-dependencies]
warp = [
    "warp-lang>=1.9.1",
]
# torch = [
#     "torch @ https://oss.mthreads.com/release-ci/computeQA/cuda_compatible/CI/sw-release/musa/external/4.3.4/deb/AI-Frameworks/torch-2.7.1-cp310-cp310-linux_x86_64.whl",
#     "torch_musa @ https://oss.mthreads.com/release-ci/computeQA/cuda_compatible/CI/sw-release/musa/external/4.3.4/deb/AI-Frameworks/torch_musa-2.7.1-cp310-cp310-linux_x86_64.whl",
# ]
tools = [
    "coverage",
    "pytest",
    "ruff",
    "black",
    "isort",
]

[project.scripts]
axinfra-unittest = "axinfra.tests.unittest_all:main"
axinfra-coverage = "axinfra.tests.run_coverage:main"

[project.urls]
Homepage = "https://sh-code.mthreads.com/sw/axInfrastructure"

[tool.setuptools]
package-data = { "axinfra" = ["bin/*.so","native/*", "native/**"] }
include-package-data = true

[tool.setuptools.packages.find]
include = ["axinfra*"]

[tool.setuptools.dynamic]
version = { file = ["VERSION.md"] }
readme = { file = ["README.md"], content-type = "text/markdown" }

[[tool.uv.index]]
name = "tuna"
url = "https://pypi.tuna.tsinghua.edu.cn/simple"
explicit = true

[[tool.uv.index]]
name = "nvidia"
url = "https://pypi.nvidia.com"
explicit = true

[tool.uv.extra-build-dependencies]
axinfra = ["setuptools","cmake>=3.18"]

[tool.uv.sources]
warp-lang = {index = "nvidia"}