mojo-gpu-puzzles/pyproject.toml at main · modular/mojo-gpu-puzzles · 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
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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
[build-system]
requires = ["setuptools>=61"]
build-backend = "setuptools.build_meta"

[project]
name = "mojo-gpu-puzzles"
version = "1.0.0"
readme = "README.md"
requires-python = ">=3.10,<3.13"
dependencies = [
    # Pinned to the release the book targets. Bump both with the release-branch
    # channel edit in `pixi.toml`; they must not drift apart again.
    "mojo==1.2.0",
    "max==26.7.0",
    "poethepoet>=0.34.0",
    "scipy>=1.15.3",
    # Matches the version `pixi.toml` pins and the suite is tested against.
    "torch==2.7.1",
]

[project.optional-dependencies]
nvidia = ["pynvml>=11.5.0"]
amd = ["pytorch-triton-rocm"]

# The repo ships no importable package — it exists to carry dependencies and
# the `poe` shortcuts. Without an explicit empty set, setuptools' flat-layout
# discovery finds `book`, `problems` and `solutions` and refuses to build.
[tool.setuptools]
py-modules = []

[[tool.uv.index]]
name = "pytorch-cu128"
url = "https://download.pytorch.org/whl/cu128"
explicit = true

[[tool.uv.index]]
name = "pytorch-rocm63"
url = "https://download.pytorch.org/whl/rocm6.3"
explicit = true

[tool.uv.sources]
# `mojo` and `max` release wheels are published to PyPI, so they need no
# dedicated index. Sourcing them from the nightly index is what pinned this
# path to an April nightly and broke it.
torch = [
  { index = "pytorch-cu128", marker = "sys_platform == 'linux'" },
]
pytorch-triton-rocm = [
  { index = "pytorch-rocm63" },
]

[tool.poe.tasks]
tests = "bash solutions/run.sh"

p01 = "mojo -I . problems/p01/p01.mojo"

p02 = "mojo -I . problems/p02/p02.mojo"

p03 = "mojo -I . problems/p03/p03.mojo"

p04 = "mojo -I . problems/p04/p04.mojo"
p04_tile_tensor = "mojo -I . problems/p04/p04_tile_tensor.mojo"
tile_tensor_intro = "mojo book/src/puzzle_04/intro.mojo"

p05 = "mojo -I . problems/p05/p05.mojo"

p06 = "mojo -I . problems/p06/p06.mojo"

p07 = "mojo -I . problems/p07/p07.mojo"

p08 = "mojo -I . problems/p08/p08.mojo"

p09 = "mojo problems/p09/p09.mojo"

p10 = "mojo problems/p10/p10.mojo"

p11 = "mojo -I . problems/p11/p11.mojo"

p12 = "mojo -I . problems/p12/p12.mojo"

p13 = "mojo -I . problems/p13/p13.mojo"

p14 = "mojo -I . problems/p14/p14.mojo"

p15 = "mojo -I . problems/p15/p15.mojo"

p16 = "bash solutions/gate.sh p16 mojo -I . problems/p16/p16.mojo"

p17 = [
    { cmd = "mojo package problems/p17/op -o problems/p17/op.mojoc" },
    { cmd = "python problems/p17/p17.py" }
]

p18-package = "mojo package problems/p18/op -o problems/p18/op.mojoc"
p18-test-kernels = { cmd = "mojo run -I problems/p18 problems/p18/test/test_softmax.mojo", deps = ["p18-package"] }
p18 = { cmd = "python problems/p18/p18.py", deps = ["p18-package"] }

p19 = [
    { cmd = "bash solutions/gate.sh p19 true" },
    { cmd = "mojo package problems/p19/op -o problems/p19/op.mojoc" },
    { cmd = "python problems/p19/p19.py" }
]

p20 = "python problems/p20/p20.py"

p21 = { cmd = "python problems/p21/p21.py", env = { PYTHONPATH = "." } }

p22 = { cmd = "bash solutions/gate.sh p22 python problems/p22/p22.py", env = { PYTHONPATH = "." } }

p23 = "mojo -I . problems/p23/p23.mojo"

p24 = "mojo -I . problems/p24/p24.mojo"

p25 = "mojo -I . problems/p25/p25.mojo"

p26 = "mojo -I . problems/p26/p26.mojo"

p27 = "mojo -I . problems/p27/p27.mojo"

p28 = "bash solutions/gate.sh p28 mojo -I . problems/p28/p28.mojo"

p29 = "bash solutions/gate.sh p29 mojo -I . problems/p29/p29.mojo"

p30 = "mojo problems/p30/p30.mojo"

p31 = "mojo problems/p31/p31.mojo"

p32 = "mojo problems/p32/p32.mojo"

p33 = "bash solutions/gate.sh p33 mojo -I . problems/p33/p33.mojo"

p34 = "bash solutions/gate.sh p34 mojo -I . problems/p34/p34.mojo"

p35 = "mojo -I . problems/p35/p35.mojo"

# GPU information
gpu-info = { cmd = "python3 scripts/gpu_specs.py --summary" }
gpu-specs = { cmd = "python3 scripts/gpu_specs.py" }