triton-viz/pyproject.toml at main · Deep-Learning-Profiling-Tools/triton-viz · 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
[tool.ruff]
ignore = ["E731"]

[build-system]
requires = ["setuptools>=42", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "triton-viz"
version = "3.0"
description = "A visualization tool for Triton"
authors = [
  {name = "Deep Learning Profiling Tools Team", email = "kzhou6@gmu.edu"}
]
readme = "README.md"
license = {text = "MIT"}
requires-python = ">=3.10"
classifiers = [
  "Programming Language :: Python :: 3",
  "License :: OSI Approved :: MIT License",
  "Operating System :: OS Independent",
]
dependencies = [
  "setuptools",
  "triton>=3.4.0",
  "pyarrow",
  "pre-commit",
  "z3-solver==4.15.3.0",
  "anytree",
  "cairocffi",
  "flask",
  "flask_cloudflared",
  "tqdm",
  "ml_dtypes",
]

[tool.setuptools.packages.find]
include=["triton_viz*"]
exclude=["tasks"]

[tool.setuptools]
include-package-data = true

[tool.setuptools.package-data]
"triton_viz" = ["templates/*.html", "static/*.js", "static/**/*.js", "static/**"]

[project.urls]
homepage = "https://github.com/Deep-Learning-Profiling-Tools/triton-viz"

[project.scripts]
triton-sanitizer = "triton_viz.wrapper:apply_sanitizer"
triton-profiler = "triton_viz.wrapper:apply_profiler"
triton-visualizer = "triton_viz.visualizer_cli:main"

[project.optional-dependencies]
test = [
  "pytest",
  "pytest-xdist",
  "numpy",
  "torch",
]
nki = [
  "neuronx-cc==2.22.12471.0+b4a00d10; python_version < '3.13'",
  "nki==0.2.0+g82fdb402; python_version < '3.13'"
]

[tool.uv]
extra-index-url = ["https://pip.repos.neuron.amazonaws.com"]
index-strategy = "unsafe-best-match"

[tool.uv.sources]
neuronx-cc = [
  {url = "https://pip.repos.neuron.amazonaws.com/neuronx-cc/neuronx_cc-2.22.12471.0%2Bb4a00d10-cp310-cp310-linux_x86_64.whl", marker = "python_version == '3.10'"},
  {url = "https://pip.repos.neuron.amazonaws.com/neuronx-cc/neuronx_cc-2.22.12471.0%2Bb4a00d10-cp311-cp311-linux_x86_64.whl", marker = "python_version == '3.11'"},
  {url = "https://pip.repos.neuron.amazonaws.com/neuronx-cc/neuronx_cc-2.22.12471.0%2Bb4a00d10-cp312-cp312-linux_x86_64.whl", marker = "python_version == '3.12'"},
]
nki = [
  {url = "https://pip.repos.neuron.amazonaws.com/nki/nki-0.2.0%2Bg82fdb402-cp310-cp310-linux_x86_64.whl", marker = "python_version == '3.10'"},
  {url = "https://pip.repos.neuron.amazonaws.com/nki/nki-0.2.0%2Bg82fdb402-cp311-cp311-linux_x86_64.whl", marker = "python_version == '3.11'"},
  {url = "https://pip.repos.neuron.amazonaws.com/nki/nki-0.2.0%2Bg82fdb402-cp312-cp312-linux_x86_64.whl", marker = "python_version == '3.12'"},
]