From 6bc95245b886b9fd2c9b3220bdb72963f525e284 Mon Sep 17 00:00:00 2001
From: Jongsok Choi <jongsok.choi@gmail.com>
Date: Sat, 4 Apr 2026 09:26:21 -0700
Subject: [PATCH] Bump minimum versions to PyTorch 2.11 and Triton 3.6

---
 .github/matrix.json        |  8 ++++----
 .github/workflows/test.yml | 10 +++++-----
 docs/installation.md       | 14 +++++++-------
 3 files changed, 16 insertions(+), 16 deletions(-)

diff --git a/.github/matrix.json b/.github/matrix.json
index 7da216a57..66f0f2173 100644
--- a/.github/matrix.json
+++ b/.github/matrix.json
@@ -7,7 +7,7 @@
       "image": "nvidia/cuda:12.8.1-devel-ubuntu24.04",
       "runtime-version": "cu128",
       "container-options": "--gpus all",
-      "pytorch-version": "pytorch-2.9",
+      "pytorch-version": "pytorch-2.11",
       "alias": "a10g",
       "backend": "triton"
     },
@@ -29,7 +29,7 @@
       "image": "nvidia/cuda:12.8.1-devel-ubuntu24.04",
       "runtime-version": "cu128",
       "container-options": "--gpus all",
-      "pytorch-version": "pytorch-2.9",
+      "pytorch-version": "pytorch-2.11",
       "alias": "a10g-dtype-asserts",
       "dtype-asserts": true,
       "expecttest-accept": true,
@@ -75,7 +75,7 @@
       "image": "nvidia/cuda:13.1.0-devel-ubuntu24.04",
       "runtime-version": "cu130",
       "container-options": "--gpus all",
-      "pytorch-version": "pytorch-2.9",
+      "pytorch-version": "pytorch-2.11",
       "alias": "b200",
       "backend": "triton"
     },
@@ -86,7 +86,7 @@
       "image": "nvidia/cuda:13.1.0-devel-ubuntu24.04",
       "runtime-version": "cu130",
       "container-options": "--gpus all",
-      "pytorch-version": "pytorch-2.9",
+      "pytorch-version": "pytorch-2.11",
       "alias": "b200",
       "backend": "tileir"
     },
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 3884781da..150f68262 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -107,9 +107,9 @@ jobs:
       - name: Install PyTorch
         run: |
           source .venv/bin/activate
-          if [[ "${{ matrix.pytorch-version }}" == "pytorch-2.9" ]]; then
-            # Install stable 2.9 from test channel
-            uv pip install -U "torch==2.9.*" --index-url https://download.pytorch.org/whl/${{ matrix.runtime-version }}
+          if [[ "${{ matrix.pytorch-version }}" == "pytorch-2.11" ]]; then
+            # Install stable 2.11 release
+            uv pip install -U "torch==2.11.*" --index-url https://download.pytorch.org/whl/${{ matrix.runtime-version }}
           elif [[ "${{ matrix.runtime-version }}" == "tpu" ]]; then
             # TPU: install CPU-only PyTorch nightly (torch_tpu provides TPU backend)
             uv pip install -U --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu
@@ -123,7 +123,7 @@ jobs:
           fi
 
       - name: Install Triton
-        if: matrix.backend == 'tileir' || (matrix.backend == 'triton' && steps.cache.outputs.cache-hit != 'true' && matrix.pytorch-version != 'pytorch-2.9')
+        if: matrix.backend == 'tileir' || (matrix.backend == 'triton' && steps.cache.outputs.cache-hit != 'true' && matrix.pytorch-version != 'pytorch-2.11')
         run: |
           set -x
           source .venv/bin/activate
@@ -269,7 +269,7 @@ jobs:
           pytest $PARALLEL -rf --timeout=60 $EXTRA_FLAGS $TEST_PATH | tee >(eval $SKIP_CHECK)
 
   test-notebooks:
-    name: test-notebooks-cu128-py3.12-pytorch-2.9-a10g
+    name: test-notebooks-cu128-py3.12-pytorch-2.11-a10g
 
     container:
       image: nvidia/cuda:12.8.1-devel-ubuntu24.04
diff --git a/docs/installation.md b/docs/installation.md
index 259aa41db..adfc59ced 100644
--- a/docs/installation.md
+++ b/docs/installation.md
@@ -13,8 +13,8 @@ Helion currently targets Linux systems and requires a recent Python and PyTorch
 - We recommend using [uv](https://docs.astral.sh/uv/) for lightweight, fast virtual environments
 
 ### Dependencies
-- **[PyTorch](https://github.com/pytorch/pytorch) 2.9 or later**
-- **[Triton](https://github.com/triton-lang/triton) 3.5 or later**
+- **[PyTorch](https://github.com/pytorch/pytorch) 2.11 or later**
+- **[Triton](https://github.com/triton-lang/triton) 3.6 or later**
 
   *Note: Older versions may work, but will lack support for features like TMA on Hopper/Blackwell GPUs and may exhibit lower performance.*
 
@@ -61,26 +61,26 @@ source .venv/bin/activate
 
 ### 2. Install PyTorch
 
-Install PyTorch 2.9 or later:
+Install PyTorch 2.11 or later:
 
 ```bash
 # CUDA 12.8
-pip install "torch==2.9.*" --index-url https://download.pytorch.org/whl/cu128
+pip install "torch==2.11.*" --index-url https://download.pytorch.org/whl/cu128
 
 # ROCm 7.0
-pip install "torch==2.9.*" --index-url https://download.pytorch.org/whl/rocm7.0
+pip install "torch==2.11.*" --index-url https://download.pytorch.org/whl/rocm7.0
 ```
 see [PyTorch installation instructions](https://pytorch.org/get-started/locally/) for other options.
 
 ### 3. Install Triton
 
-Install Triton 3.5 or later using one of the options below.
+Install Triton 3.6 or later using one of the options below.
 
 #### Option A: Prebuilt wheel (recommended)
 
 ```bash
 # Install Triton from PyPI (will pick the right wheel for your platform if available)
-pip install "triton>=3.5"
+pip install "triton>=3.6"
 
 # Verify
 python -c "import triton; print('Triton version:', triton.__version__)"