From f2b91b03ca51d01f93b573e342a01ca898fe67ab Mon Sep 17 00:00:00 2001 From: bowenyang008 Date: Thu, 29 May 2025 19:07:09 +0000 Subject: [PATCH 1/5] version bump --- .github/workflows/pr-gpu.yaml | 8 ++++---- pyproject.toml | 2 +- setup.py | 6 +++--- 3 files changed, 8 insertions(+), 8 deletions(-) diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml index 4e462095..758323d4 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.6.0-gpus1" + - name: "python3.11-pytorch2.7.0-gpus1" gpu_num: 1 python_version: 3.11 - container: mosaicml/pytorch:2.6.0_cu124-python3.11-ubuntu22.04 - - name: "python3.11-pytorch2.6.0-gpus2" + container: mosaicml/pytorch:2.7.0_cu126-python3.11-ubuntu22.04 + - name: "python3.11-pytorch2.7.0-gpus2" gpu_num: 2 python_version: 3.11 - container: mosaicml/pytorch:2.6.0_cu124-python3.11-ubuntu22.04 + container: mosaicml/pytorch:2.7.0_cu126-python3.11-ubuntu22.04 steps: - name: Run PR GPU tests uses: mosaicml/ci-testing/.github/actions/pytest-gpu@v0.3.5 diff --git a/pyproject.toml b/pyproject.toml index 54d4f5a2..d8471105 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -3,7 +3,7 @@ # build requirements [build-system] -requires = ["setuptools < 79.0.0", "torch >= 2.6.0, < 2.6.1"] +requires = ["setuptools < 79.0.0", "torch >= 2.7.0, < 2.7.1"] build-backend = "setuptools.build_meta" # Pytest diff --git a/setup.py b/setup.py index 1b80f17c..c62797fe 100644 --- a/setup.py +++ b/setup.py @@ -62,15 +62,15 @@ install_requires = [ 'numpy>=1.21.5,<2.1.0', 'packaging>=21.3.0,<24.2', - 'torch>=2.6.0,<2.6.1', - 'triton>=3.2.0,<3.3.0', + 'torch>=2.7.0,<2.7.1', + 'triton>=3.3.0,<3.4.0', 'stanford-stk==0.7.1', ] extra_deps = {} extra_deps['gg'] = [ - 'grouped_gemm==0.2.0', + 'grouped_gemm==0.3.0', ] extra_deps['dev'] = [ From b8c73d459855d46288048779c765a12fdf180cd4 Mon Sep 17 00:00:00 2001 From: bowenyang008 Date: Thu, 29 May 2025 22:20:23 +0000 Subject: [PATCH 2/5] use pull_request instead --- .github/workflows/pr-gpu.yaml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml index 758323d4..d196d9d2 100644 --- a/.github/workflows/pr-gpu.yaml +++ b/.github/workflows/pr-gpu.yaml @@ -4,7 +4,7 @@ on: branches: - main - release/* - pull_request_target: + pull_request: branches: - main - release/** @@ -24,11 +24,11 @@ jobs: - name: "python3.11-pytorch2.7.0-gpus1" gpu_num: 1 python_version: 3.11 - container: mosaicml/pytorch:2.7.0_cu126-python3.11-ubuntu22.04 + container: mosaicml/pytorch:2.7.0_cu128-python3.11-ubuntu22.04 - name: "python3.11-pytorch2.7.0-gpus2" gpu_num: 2 python_version: 3.11 - container: mosaicml/pytorch:2.7.0_cu126-python3.11-ubuntu22.04 + container: mosaicml/pytorch:2.7.0_cu128-python3.11-ubuntu22.04 steps: - name: Run PR GPU tests uses: mosaicml/ci-testing/.github/actions/pytest-gpu@v0.3.5 From 4dbf8a08a5a450c3e56e21516290eaf2e2685fdc Mon Sep 17 00:00:00 2001 From: bowenyang008 Date: Thu, 29 May 2025 22:37:46 +0000 Subject: [PATCH 3/5] python 3.12 --- .github/workflows/pr-gpu.yaml | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml index d196d9d2..543e6d78 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.7.0-gpus1" + - name: "python3.12-pytorch2.7.0-gpus1" gpu_num: 1 - python_version: 3.11 - container: mosaicml/pytorch:2.7.0_cu128-python3.11-ubuntu22.04 - - name: "python3.11-pytorch2.7.0-gpus2" + python_version: 3.12 + container: mosaicml/pytorch:2.7.0_cu128-python3.12-ubuntu22.04 + - name: "python3.12-pytorch2.7.0-gpus2" gpu_num: 2 - python_version: 3.11 - container: mosaicml/pytorch:2.7.0_cu128-python3.11-ubuntu22.04 + python_version: 3.12 + container: mosaicml/pytorch:2.7.0_cu128-python3.12-ubuntu22.04 steps: - name: Run PR GPU tests uses: mosaicml/ci-testing/.github/actions/pytest-gpu@v0.3.5 From 953746423f7c8dd724a6d80f1c38409c2c8deffe Mon Sep 17 00:00:00 2001 From: bowenyang008 Date: Thu, 29 May 2025 22:50:33 +0000 Subject: [PATCH 4/5] rm triton --- setup.py | 1 - 1 file changed, 1 deletion(-) diff --git a/setup.py b/setup.py index c62797fe..e625235e 100644 --- a/setup.py +++ b/setup.py @@ -63,7 +63,6 @@ 'numpy>=1.21.5,<2.1.0', 'packaging>=21.3.0,<24.2', 'torch>=2.7.0,<2.7.1', - 'triton>=3.3.0,<3.4.0', 'stanford-stk==0.7.1', ] From 01092db91db789d03e59592e53618e4813b5bd75 Mon Sep 17 00:00:00 2001 From: bowenyang008 Date: Thu, 29 May 2025 23:14:55 +0000 Subject: [PATCH 5/5] revert to target --- .github/workflows/pr-gpu.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/pr-gpu.yaml b/.github/workflows/pr-gpu.yaml index 543e6d78..a94bd1f9 100644 --- a/.github/workflows/pr-gpu.yaml +++ b/.github/workflows/pr-gpu.yaml @@ -4,7 +4,7 @@ on: branches: - main - release/* - pull_request: + pull_request_target: branches: - main - release/**