magnetron/pyproject.toml at master · MarioSieg/magnetron · 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
83
84
85
86
87
88
89
90
91
[build-system]
requires = [
    "scikit-build-core"
]
build-backend = "scikit_build_core.build"

[project]
name = "magnetron"
version = "0.1.6"
description = "A compact machine learning runtime for developers who want to understand, control, and optimize the full stack."
requires-python = ">=3.10"
readme = "README.md"
license = { file = "LICENSE" }
authors = [{ name = "Mario Sieg", email = "mario.sieg.64@gmail.com" }]
dependencies = []
keywords = [
    "machine-learning",
    "deep-learning",
    "neural-networks",
    "gpu",
    "simd",
]

[urls]
Homepage = "https://github.com/MarioSieg/magnetron"
Source = "https://github.com/MarioSieg/magnetron"
Issues = "https://github.com/MarioSieg/magnetron/issues"

[project.optional-dependencies]
examples = [
    "tokenizers",
    "rich",
    "matplotlib",
    "huggingface_hub"
]

test = [
    "ruff",
    "torch",
    "numpy",
    "transformers",
    "tiktoken",
    "matplotlib",
    "pytest",
    "pytest-xdist",
    "pyinstrument",
    "rich"
]

[tool.scikit-build.cmake.define]
# Enable / disable backends
MAGNETRON_ENABLE_BACKEND_CPU = true
MAGNETRON_ENABLE_BACKEND_CUDA = false

# Enable / disable features
MAGNETRON_BUILD_PYTHON_BINDINGS = true # Required for the Python package
MAGNETRON_DEBUG = false
MAGNETRON_BUILD_TESTS = false
MAGNETRON_BUILD_BENCHMARKS = false

[tool.scikit-build]
wheel.packages = ["python/magnetron"]
editable.mode = "redirect"
install.components = ["python"]
build-dir = "build/{wheel_tag}"

[tool.ruff]
line-length = 150
exclude = [
    ".venv",
    "docs",
    "extern",
    "test",
    "tools",
    "fuzzer",
    "cmake",
    "bin",
]

[tool.ruff.lint]
ignore = ["F403"]
select = [
    "ANN",
    "UP"
]

[tool.ruff.format]
quote-style = "single"

[tool.ruff.lint.per-file-ignores]
"setup.py" = ["ANN"]