Skip to content

Commit 9634e20

Browse files
committed
Update pixi.toml to include all environments.
1 parent 563f4e0 commit 9634e20

File tree

2 files changed

+57
-35
lines changed

2 files changed

+57
-35
lines changed

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
/git/
2+
/.zed/
23

34

45
# pixi environments

pixi.toml

Lines changed: 56 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -2,14 +2,14 @@
22
authors = ["Hameer Abbasi <[email protected]>"]
33
channels = ["conda-forge"]
44
name = "pytorch-dev"
5-
platforms = ["linux-64"]
5+
platforms = ["linux-64", "osx-arm64"]
66
version = "0.1.0"
77

88
[tasks]
9-
build-core = "./pytorch-build.sh"
10-
build-all = "./torch-build.sh"
11-
update = "./torch-update.sh"
12-
clone = "./torch-clone.sh"
9+
build-core = { cmd = "./pytorch-build.sh", cwd = "." }
10+
build-all = { cmd = "./torch-build.sh", cwd = "." }
11+
update = { cmd = "./torch-update.sh", cwd = "." }
12+
clone = { cmd = "./torch-clone.sh", cwd = "." }
1313

1414
[activation.env]
1515
PYTORCH_BUILD_DIRECTORY = "${PIXI_PROJECT_ROOT}/git"
@@ -18,43 +18,38 @@ PYTORCH_PIXI_BUILD = "1"
1818
[activation]
1919
scripts = ["./torch-common.sh"]
2020

21+
[feature.freethreading.dependencies]
22+
python = "3.13.*"
23+
python-freethreading = "3.13.*"
24+
2125
[dependencies]
2226
python = "3.9.*"
27+
28+
[target.linux-64.dependencies]
29+
# These three don't have free-threading support
30+
lldb = "*"
31+
gdb = "*"
32+
halide-python = "*"
33+
34+
[feature.core.dependencies]
2335
pip = "*"
2436
cpython = "*"
25-
gcc = "13.*"
26-
gxx = "13.*"
27-
gdb = "*"
28-
clang = "*"
29-
clangxx = "*"
30-
lldb = "*"
31-
magma = "*"
32-
cuda-driver-dev = "*"
33-
cuda-version = "12.6.*"
34-
cudnn = "*"
35-
conda-gcc-specs = "*"
36-
cuda-libraries-dev = "*"
37-
cuda-nvcc = "*"
38-
cuda-gdb = "*"
39-
cuda-nvtx-dev = "*"
40-
cuda-nvml-dev = "*"
41-
cuda-cupti-dev = "*"
42-
ccache = "*"
43-
cmake = "*"
44-
fsspec = "*"
45-
mkl = "*"
46-
mkl-include = "*"
37+
cmake = ">=3.27"
4738
ninja = "*"
39+
numpy = "*"
4840
packaging = "*"
41+
pkg-config = "*"
42+
libuv = "*"
4943
pyyaml = "*"
44+
requests = "*"
45+
six = "*"
46+
typing-extensions = ">=4.10.0"
47+
ccache = "*"
5048
scikit-build = "*"
51-
setuptools = "*"
52-
sysroot_linux-64 = ">=2.17"
53-
types-dataclasses = "*"
54-
typing = "*"
55-
typing-extensions = "*"
49+
setuptools = ">=70.1.0,<80.0a0"
5650
wheel = "*"
57-
numpy = "*"
51+
fsspec = "*"
52+
types-dataclasses = "*"
5853
expecttest = "*"
5954
hypothesis = "*"
6055
lintrunner = "*"
@@ -75,11 +70,37 @@ optree = "*"
7570
tabulate = "*"
7671
networkx = "*"
7772
zlib = "*"
78-
halide-python = "*"
7973
parameterized = "*"
8074
libglib = "*"
8175
pango = "*"
8276
pillow = "*"
8377

84-
[pypi-dependencies]
78+
[feature.core.target.linux-64.dependencies]
79+
gcc = "13.*"
80+
gxx = "13.*"
81+
clang = "*"
82+
clangxx = "*"
83+
magma = "*"
84+
cuda-driver-dev = "*"
85+
cuda-version = "12.6.*"
86+
cudnn = "*"
87+
conda-gcc-specs = "*"
88+
cuda-libraries-dev = "*"
89+
cuda-nvcc = "*"
90+
cuda-gdb = "*"
91+
cuda-nvtx-dev = "*"
92+
cuda-nvml-dev = "*"
93+
cuda-cupti-dev = "*"
94+
mkl = "*"
95+
mkl-include = "*"
96+
sysroot_linux-64 = ">=2.17"
97+
98+
[feature.core.pypi-dependencies]
8599
z3-solver = "*"
100+
101+
[environments]
102+
default = ["default", "core"]
103+
freethreading = { features = [
104+
"freethreading",
105+
"core",
106+
], no-default-feature = true }

0 commit comments

Comments
 (0)