-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy pathpyproject.toml
119 lines (100 loc) · 3.4 KB
/
pyproject.toml
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
[project]
name = "snax-mlir"
description = "Driving SNAX with MLIR"
version = "0.2.2"
requires-python = "== 3.12.*"
dependencies = [
"xdsl @ git+https://github.com/xdslproject/xdsl.git@e9b1efb65d0873ee21442515e78f3a449b01bedb",
"numpy",
]
[project.optional-dependencies]
# All optional dependencies for plotting and parsing
dev = [
"pre-commit",
"filecheck==1.0.0",
"lit",
"pytest",
]
# All optional dependencies for graphical plotting
viz = [
"pandas",
"matplotlib",
]
# Dependencies for neural network compilation
nn = [
"tensorflow-cpu==2.16.2",
]
[project.scripts]
snax-opt = "snaxc.tools.snax_opt_main:main"
[tool.ruff]
select = ["E", "F", "W", "I", "UP", "PT", "TID251"]
ignore = [
"E741", # https://beta.ruff.rs/docs/rules/ambiguous-variable-name/
"PT006", # https://beta.ruff.rs/docs/rules/pytest-parametrize-names-wrong-type/
"PT007", # https://beta.ruff.rs/docs/rules/pytest-parametrize-values-wrong-type/
"PT011", # https://beta.ruff.rs/docs/rules/pytest-raises-too-broad/
"PT012", # https://beta.ruff.rs/docs/rules/pytest-raises-with-multiple-statements/
"PT015", # https://beta.ruff.rs/docs/rules/pytest-assert-always-false/
]
target-version = "py310"
line-length = 120
[tool.ruff.flake8-tidy-imports.banned-api]
"xdsl.parser.core".msg = "Use xdsl.parser instead."
"xdsl.parser.attribute_parser".msg = "Use xdsl.parser instead."
"xdsl.parser.affine_parser".msg = "Use xdsl.parser instead."
"xdsl.ir.core".msg = "Use xdsl.ir instead."
"xdsl.irdl.irdl".msg = "Use xdsl.irdl instead"
"xdsl.ir.affine.affine_expr".msg = "Use xdsl.ir.affine instead"
"xdsl.ir.affine.affine_map".msg = "Use xdsl.ir.affine instead"
"xdsl.ir.affine.affine_set".msg = "Use xdsl.ir.affine instead"
[tool.ruff.per-file-ignores]
"__init__.py" = ["F403"]
"tests/filecheck/frontend/programs/invalid.py" = ["F811", "F841"]
"tests/filecheck/frontend/dialects/invalid.py" = ["F811"]
"tests/test_declarative_assembly_format.py" = ["F811"]
"versioneer.py" = ["ALL"]
"_version.py" = ["ALL"]
[tool.ruff.mccabe]
max-complexity = 10
[tool.pyright]
reportImportCycles = false
reportMissingModuleSource = false
enableTypeIgnoreComments = false
enableExperimentalFeatures = true
typeCheckingMode = "strict"
"include" = [
"snaxc",
"tests",
]
"exclude" = [
"snaxc/dialects/accfg.py",
"snaxc/inference/dataflow.py",
"snaxc/inference/helpers.py",
"snaxc/inference/scoped_setups.py",
"snaxc/inference/trace_acc_state.py",
"snaxc/transforms/accfg_dedup.py",
"snaxc/transforms/convert_linalg_to_accfg.py",
"snaxc/transforms/frontend/preprocess_mlperf_tiny.py",
"snaxc/transforms/reuse_memref_allocs.py",
"snaxc/transforms/set_memory_layout.py",
"snaxc/transforms/snax_copy_to_dma.py",
"tests/benchmark/test_snax_benchmark.py",
"tests/inference/test_accfg_state_tracing.py",
"tests/util/",
]
[tool.pixi.project]
channels = ["conda-forge", "bioconda", "kuleuven-micas"]
platforms = ["linux-64"]
[tool.pixi.pypi-dependencies]
snax-mlir = { path = ".", editable = true }
[tool.pixi.environments]
default = { features = ["dev", "nn", "viz"] }
[tool.pixi.tasks]
[tool.pixi.dependencies]
mlir = { version = "==19.1.1.c.d401987fe349a87c53fe25829215b80b70c0c1a", channel = "kuleuven-micas" }
clang = "==19.1.1"
lld = "==19.1.1"
snax-cluster-prebuilt = "==0.2.12"
snakemake-minimal="*"
pyright = ">=1.1.390,<2"
vcd-to-csv = { version = ">=0.1.0,<0.2", channel = "kuleuven-micas" }