tt-metal/pyproject.toml at main · kc-ml2/tt-metal · GitHub

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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
[build-system]
requires = [
  "setuptools==70.1.0",
  "setuptools-scm==8.1.0",
  "wheel",
]
build-backend = "setuptools.build_meta"

[project]
name = "ttnn"
authors = [
    {name = "Tenstorrent"},
    {email = "info@tenstorrent.com"},
]
dependencies = [
  "numpy>=1.24.4,<2",
  "loguru>=0.6.0",
  "networkx>=3.1",
  "graphviz>=0.20.3",
  "pyyaml>=5.4",
  "click>=8.1.7",
  "pandas>=2.0.3",
  "seaborn>=0.13.2",
  "ml_dtypes",
]
requires-python = ">=3.10"
keywords = ["metalium", "ttnn", "tt-metal", "machine learning", "tenstorrent", "libnuma", "libhwloc"]
license = { text = "Apache-2.0" }
description = "General compute framework for Tenstorrent devices"
dynamic = ["version", "readme"]

[project.urls]
Homepage = "https://docs.tenstorrent.com/tt-metal/latest/ttnn"
Documentation = "https://docs.tenstorrent.com/tt-metal/latest/ttnn"
Repository = "https://github.com/tenstorrent/tt-metal"
Issues = "https://github.com/tenstorrent/tt-metal/issues"

[project.scripts]
tt-run = "ttnn.distributed.ttrun:main"

[tool.black]
line-length = 120
include = '^.*(ttnn|tests/scripts|tests/ttnn|tests/tt_eager/python_api_testing|tt_eager/tt_lib|tests/scripts|models/demos|infra|.github)/.*\.py$'

[tool.isort]
profile = "black"
line_length = 120
skip_gitignore = true
skip = ["__init__.py"]

[tool.ruff]
line-length = 120

[tool.setuptools_scm]
# Supported tag formats:
# v1.2.3, v1.2.3-rc1
tag_regex = '^(?:v)?(?P<version>\d+\.\d+\.\d+(?:-rc\d+)?)$'
#Only look for release tags for calculating distance for wheel name
git_describe_command = [
  "git","describe","--dirty","--tags","--long",
  "--match","v[0-9]*.[0-9]*.[0-9]*",
  "--match","v[0-9]*.[0-9]*.[0-9]*-rc[0-9]*",
  "--exclude","*-dev*"
]
fallback_version = "0.0.0.dev0"


[tool.cibuildwheel]
skip = "*-musllinux_*"

[tool.uv]
# Use "copy" mode instead of the default "hardlink" or "symlink" modes.
# This ensures packages are fully independent copies in the virtual environment,
# which is more compatible with:
# - Network filesystems (NFS, CIFS) that may not support hardlinks across mount points
# - Container environments where the cache and venv may be on different volumes
# - Scenarios where the uv cache might be cleared independently of the venv
# Trade-off: Uses more disk space but provides better isolation and portability.
#
# To change: Set to "hardlink" for better disk space usage on local filesystems,
# or "symlink" for fastest operations (requires same filesystem for cache and venv).
link-mode = "copy"