Add freethreading support

Quansight · Oct 16, 2024 · 88041a6 · 88041a6
1 parent 7d51de1
commit 88041a6
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/pytorch-dev.yaml b/pytorch-dev.yaml
@@ -3,7 +3,8 @@ channels:
   - conda-forge
   - nodefaults
 dependencies:
-  - python=3.9
+  - python=3.13
+  - python-freethreading
   - pip
   - cpython
 
@@ -36,7 +37,7 @@ dependencies:
   - mkl-include
   - ninja
   - packaging
-  - pyyaml
+  #- pyyaml
   - scikit-build
   - setuptools
   - sysroot_linux-64>=2.17
@@ -50,12 +51,12 @@ dependencies:
   - expecttest
   - hypothesis
   - lintrunner
-  - scipy
+  #- scipy
   - psutil
   - pytest
 
   # Development utilities
-  - ghstack
+  #- ghstack
   - ipython
   - pytest-sugar
   - pytest-xdist
@@ -69,17 +70,18 @@ dependencies:
   - dill
   - jinja2
   - lark
-  - optree
+  #- optree
   - tabulate
   - networkx
   - pip:
+    - pyyaml
     - z3-solver
 
   # triton dependencies, if building from source
   - zlib
 
   # torch.compile halide backend
-  - halide-python
+  #- halide-python
 
   # torchbench dependencies that may cause issues if installed by pip
   # - pandas