From 0380cce59b06d6d2d726b21b1f24b0549f3ac42c Mon Sep 17 00:00:00 2001 From: Chuck Tang Date: Mon, 18 Nov 2024 15:27:15 -0800 Subject: [PATCH 1/4] commit change --- .github/workflows/pr-gpu.yaml | 8 ++++---- megablocks/_version.py | 2 +- setup.py | 2 +- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml index d94b057..dda6e68 100644 --- 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 index 5f259c3..5148393 100644 --- 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/setup.py b/setup.py index 4b7be37..5338d6b 100644 --- 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', ] From 9ff080c2137e650feeb34096c45a0d9f1aa9e9d4 Mon Sep 17 00:00:00 2001 From: Chuck Tang Date: Mon, 18 Nov 2024 15:33:15 -0800 Subject: [PATCH 2/4] commit change --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index ad4dc1b..28b9135 100644 --- 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 From 875a19ac933135ecaf945bdb32be691929192b57 Mon Sep 17 00:00:00 2001 From: Chuck Tang Date: Mon, 18 Nov 2024 15:56:34 -0800 Subject: [PATCH 3/4] commit change --- megablocks/_version.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/megablocks/_version.py b/megablocks/_version.py index 5148393..5f259c3 100644 --- a/megablocks/_version.py +++ b/megablocks/_version.py @@ -3,4 +3,4 @@ """The MegaBlocks Version.""" -__version__ = '0.8.0.dev0' +__version__ = '0.7.0.dev0' From 19e276be11536412329fc54113227b2cc7de62c8 Mon Sep 17 00:00:00 2001 From: Chuck Tang Date: Tue, 19 Nov 2024 15:31:11 -0800 Subject: [PATCH 4/4] commit change --- megablocks/_version.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/megablocks/_version.py b/megablocks/_version.py index 5f259c3..5148393 100644 --- 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'