Bump torch 2.5.1 + megablocks version upgrade (#162)

j316chuck · web-flow · commit 75a2560b8524 · 2024-11-19T16:02:39.000-08:00
diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml
@@ -21,14 +21,14 @@ jobs:
       fail-fast: false
       matrix:
         include:
-        - name: "python3.11-pytorch2.4.0-gpus1"
+        - name: "python3.11-pytorch2.5.1-gpus1"
           gpu_num: 1
           python_version: 3.11
-          container: mosaicml/pytorch:2.4.0_cu124-python3.11-ubuntu20.04
-        - name: "python3.11-pytorch2.4.0-gpus2"
+          container: mosaicml/pytorch:2.5.1_cu124-python3.11-ubuntu20.04
+        - name: "python3.11-pytorch2.5.1-gpus2"
           gpu_num: 2
           python_version: 3.11
-          container: mosaicml/pytorch:2.4.0_cu124-python3.11-ubuntu20.04
+          container: mosaicml/pytorch:2.5.1_cu124-python3.11-ubuntu20.04
     steps:
     - name: Run PR GPU tests
       uses: mosaicml/ci-testing/.github/actions/pytest-gpu@v0.1.2
diff --git a/megablocks/_version.py b/megablocks/_version.py
@@ -3,4 +3,4 @@
 
 """The MegaBlocks Version."""
 
-__version__ = '0.7.0.dev0'
+__version__ = '0.8.0.dev0'
diff --git a/pyproject.toml b/pyproject.toml
@@ -3,7 +3,7 @@
 
 # build requirements
 [build-system]
-requires = ["setuptools < 70.0.0", "torch >= 2.4.0, < 2.4.1"]
+requires = ["setuptools < 70.0.0", "torch >= 2.5.1, < 2.5.2"]
 build-backend = "setuptools.build_meta"
 
 # Pytest
diff --git a/setup.py b/setup.py
@@ -62,7 +62,7 @@
 install_requires = [
     'numpy>=1.21.5,<2.1.0',
     'packaging>=21.3.0,<24.2',
-    'torch>=2.4.0,<2.4.1',
+    'torch>=2.5.1,<2.5.2',
     'triton>=2.1.0',
     'stanford-stk==0.7.1',
 ]

Original file line number	Diff line number	Diff line change
`@@ -3,4 +3,4 @@`
`3`	`3`
`4`	`4`	`"""The MegaBlocks Version."""`
`5`	`5`
`6`		`-__version__ = '0.7.0.dev0'`
	`6`	`+__version__ = '0.8.0.dev0'`
Original file line number	Diff line number	Diff line change
`@@ -62,7 +62,7 @@`
`62`	`62`	`install_requires = [`
`63`	`63`	`'numpy>=1.21.5,<2.1.0',`
`64`	`64`	`'packaging>=21.3.0,<24.2',`
`65`		`- 'torch>=2.4.0,<2.4.1',`
	`65`	`+ 'torch>=2.5.1,<2.5.2',`
`66`	`66`	`'triton>=2.1.0',`
`67`	`67`	`'stanford-stk==0.7.1',`
`68`	`68`	`]`