From eef607ee8b1a551e9daffdb26a89ebc7f96395fa Mon Sep 17 00:00:00 2001 From: Liron Kesem Date: Fri, 21 Nov 2025 14:33:49 -0500 Subject: [PATCH 1/9] first try pipeline --- .github/workflows/test.yaml | 70 +++++++++++++++++++++++++++++++++++++ 1 file changed, 70 insertions(+) create mode 100644 .github/workflows/test.yaml diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml new file mode 100644 index 0000000..c1c5795 --- /dev/null +++ b/.github/workflows/test.yaml @@ -0,0 +1,70 @@ +name: TritonBench Verification + +on: + push: + branches: + - main + pull_request: + +jobs: + cuda: + name: CUDA Verification + runs-on: ubuntu-22.04 + strategy: + matrix: + python-version: [3.10] + + steps: + - name: Checkout code + uses: actions/checkout@v3 + + - name: Setup Python + uses: actions/setup-python@v4 + with: + python-version: ${{ matrix.python-version }} + + - name: Install system dependencies (CUDA) + run: | + sudo apt-get update + sudo apt-get install -y build-essential cmake git wget python3-pip + + - name: Install Python dependencies & setup kernelGeneration (CUDA) + run: | + pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu124 + python kernelGeneration/install.py + + - name: Run TritonBench test (CUDA) + run: | + python tritonbench/run.py --op bf16_layernorm + + amd: + name: AMD Verification + runs-on: ubuntu-22.04 + strategy: + matrix: + python-version: [3.10] + + steps: + - name: Checkout code + uses: actions/checkout@v3 + + - name: Setup Python + uses: actions/setup-python@v4 + with: + python-version: ${{ matrix.python-version }} + + - name: Install system dependencies (ROCm) + run: | + sudo apt-get update + sudo apt-get install -y build-essential cmake git wget python3-pip + + + - name: Install Python dependencies & setup kernelGeneration (ROCm) + run: | + pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.4 + python kernelGeneration/install.py + + - name: Run TritonBench test (ROCm) + run: | + export TRITON_F32_DEFAULT=1 + python tritonbench/run.py --op bf16_layernorm From d8f88aa98a7c354239a563be8c09720f1c597aa5 Mon Sep 17 00:00:00 2001 From: Liron Kesem Date: Fri, 21 Nov 2025 14:41:48 -0500 Subject: [PATCH 2/9] trying 2 --- .github/workflows/test.yaml | 1 + 1 file changed, 1 insertion(+) diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index c1c5795..8f9f398 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -4,6 +4,7 @@ on: push: branches: - main + - dev pull_request: jobs: From 177a89b3a031845bb6141c455a56fa9133f813b4 Mon Sep 17 00:00:00 2001 From: Liron Kesem Date: Fri, 21 Nov 2025 14:51:26 -0500 Subject: [PATCH 3/9] try3 --- .github/workflows/test.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index 8f9f398..29fc0cc 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -13,7 +13,7 @@ jobs: runs-on: ubuntu-22.04 strategy: matrix: - python-version: [3.10] + python-version: ["3.10"] steps: - name: Checkout code @@ -22,7 +22,7 @@ jobs: - name: Setup Python uses: actions/setup-python@v4 with: - python-version: ${{ matrix.python-version }} + python-version: ${{ matrix.python-version }} - name: Install system dependencies (CUDA) run: | From 5e7ab505027a69a0899a68ad3836254c889a4fe7 Mon Sep 17 00:00:00 2001 From: Liron Kesem Date: Fri, 21 Nov 2025 15:08:07 -0500 Subject: [PATCH 4/9] fixing the torch for nvidia --- .github/workflows/test.yaml | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index 29fc0cc..7b0f9e3 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -13,7 +13,7 @@ jobs: runs-on: ubuntu-22.04 strategy: matrix: - python-version: ["3.10"] + python-version: ['3.10'] steps: - name: Checkout code @@ -22,7 +22,7 @@ jobs: - name: Setup Python uses: actions/setup-python@v4 with: - python-version: ${{ matrix.python-version }} + python-version: 3.10 - name: Install system dependencies (CUDA) run: | @@ -31,7 +31,7 @@ jobs: - name: Install Python dependencies & setup kernelGeneration (CUDA) run: | - pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu124 + pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu126 python kernelGeneration/install.py - name: Run TritonBench test (CUDA) @@ -43,7 +43,7 @@ jobs: runs-on: ubuntu-22.04 strategy: matrix: - python-version: [3.10] + python-version: ['3.10'] steps: - name: Checkout code @@ -52,7 +52,7 @@ jobs: - name: Setup Python uses: actions/setup-python@v4 with: - python-version: ${{ matrix.python-version }} + python-version: 3.10 - name: Install system dependencies (ROCm) run: | From 6e5b599d3d963317947c4770e7fc8e3d16522010 Mon Sep 17 00:00:00 2001 From: Liron Kesem Date: Fri, 21 Nov 2025 15:14:00 -0500 Subject: [PATCH 5/9] try --- .github/workflows/test.yaml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index 7b0f9e3..a3e5607 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -13,7 +13,7 @@ jobs: runs-on: ubuntu-22.04 strategy: matrix: - python-version: ['3.10'] + python-version: [3.10] steps: - name: Checkout code @@ -22,7 +22,7 @@ jobs: - name: Setup Python uses: actions/setup-python@v4 with: - python-version: 3.10 + python-version: ${{ matrix.python-version }} - name: Install system dependencies (CUDA) run: | @@ -52,7 +52,7 @@ jobs: - name: Setup Python uses: actions/setup-python@v4 with: - python-version: 3.10 + python-version: ${{ matrix.python-version }} - name: Install system dependencies (ROCm) run: | From f1f059b6742e33d4e7f228bf59eb76e668d75ffd Mon Sep 17 00:00:00 2001 From: Liron Kesem Date: Fri, 21 Nov 2025 15:16:29 -0500 Subject: [PATCH 6/9] fixing Cuda py version --- .github/workflows/test.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index a3e5607..263619d 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -13,7 +13,7 @@ jobs: runs-on: ubuntu-22.04 strategy: matrix: - python-version: [3.10] + python-version: ['3.10'] steps: - name: Checkout code From 56de2db4d4a5e12d69c9d24c3ad6998e40aee302 Mon Sep 17 00:00:00 2001 From: Liron Kesem Date: Fri, 21 Nov 2025 15:19:37 -0500 Subject: [PATCH 7/9] trying only amd+adding --no-cache-dir --- .github/workflows/test.yaml | 32 +------------------------------- 1 file changed, 1 insertion(+), 31 deletions(-) diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index 263619d..d6633a8 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -8,36 +8,6 @@ on: pull_request: jobs: - cuda: - name: CUDA Verification - runs-on: ubuntu-22.04 - strategy: - matrix: - python-version: ['3.10'] - - steps: - - name: Checkout code - uses: actions/checkout@v3 - - - name: Setup Python - uses: actions/setup-python@v4 - with: - python-version: ${{ matrix.python-version }} - - - name: Install system dependencies (CUDA) - run: | - sudo apt-get update - sudo apt-get install -y build-essential cmake git wget python3-pip - - - name: Install Python dependencies & setup kernelGeneration (CUDA) - run: | - pip3 install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu126 - python kernelGeneration/install.py - - - name: Run TritonBench test (CUDA) - run: | - python tritonbench/run.py --op bf16_layernorm - amd: name: AMD Verification runs-on: ubuntu-22.04 @@ -62,7 +32,7 @@ jobs: - name: Install Python dependencies & setup kernelGeneration (ROCm) run: | - pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.4 + pip install --pre --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.4 python kernelGeneration/install.py - name: Run TritonBench test (ROCm) From b73d3e3bbf75035b174bcc79abe8d6771acae49c Mon Sep 17 00:00:00 2001 From: Liron Kesem Date: Fri, 21 Nov 2025 15:27:27 -0500 Subject: [PATCH 8/9] trying with whl files --- .github/workflows/test.yaml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index d6633a8..1fcffac 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -32,7 +32,8 @@ jobs: - name: Install Python dependencies & setup kernelGeneration (ROCm) run: | - pip install --pre --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.4 + pip download torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/rocm6.4 -d whls/rocm + pip install --no-index --find-links=whls/rocm torch torchvision python kernelGeneration/install.py - name: Run TritonBench test (ROCm) From b6033d101db329ec613ff5e2860b27fcbd68ec30 Mon Sep 17 00:00:00 2001 From: Liron Kesem Date: Fri, 21 Nov 2025 16:09:27 -0500 Subject: [PATCH 9/9] not fixing torch ver --- .github/workflows/test.yaml | 39 ++++++++++++++++++++++++++++++++++--- 1 file changed, 36 insertions(+), 3 deletions(-) diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index 1fcffac..6a6add5 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -13,9 +13,26 @@ jobs: runs-on: ubuntu-22.04 strategy: matrix: - python-version: ['3.10'] + python-version: ['3.10'] steps: + - name: Show memory and disk (start) (Debug) + run: | + echo "=== MEMORY ===" + free -h + echo "=== DISK ===" + df -h + + - name: Free disk space + run: | + sudo rm -rf /usr/local/lib/android + sudo rm -rf /usr/share/dotnet + sudo rm -rf /opt/ghc + sudo rm -rf /opt/hostedtoolcache/CodeQL + sudo docker image prune -af + sudo apt-get clean + sudo rm -rf /var/lib/apt/lists/* + - name: Checkout code uses: actions/checkout@v3 @@ -29,12 +46,28 @@ jobs: sudo apt-get update sudo apt-get install -y build-essential cmake git wget python3-pip + - name: Show memory & disk (before PyTorch install) (Debug) + run: | + echo "=== MEMORY ===" + free -h + echo "=== DISK ===" + df -h + + - name: show repo tree (Debug) + run: ls -R . - name: Install Python dependencies & setup kernelGeneration (ROCm) run: | - pip download torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/rocm6.4 -d whls/rocm + pip download torch torchvision --index-url https://download.pytorch.org/whl/nightly/rocm6.4 -d whls/rocm pip install --no-index --find-links=whls/rocm torch torchvision - python kernelGeneration/install.py + python install.py + + - name: Show memory & disk (after PyTorch install) (Debug) + run: | + echo "=== MEMORY ===" + free -h + echo "=== DISK ===" + df -h - name: Run TritonBench test (ROCm) run: |