From 3cdd5b3fe6629472cebea31915dc709bcac03913 Mon Sep 17 00:00:00 2001 From: Oguz Ulgen Date: Tue, 7 Oct 2025 16:25:31 -0700 Subject: [PATCH] Match cuda versions for benchmark --- .github/workflows/benchmark_dispatch.yml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/benchmark_dispatch.yml b/.github/workflows/benchmark_dispatch.yml index 011d9e891..16dd1712f 100644 --- a/.github/workflows/benchmark_dispatch.yml +++ b/.github/workflows/benchmark_dispatch.yml @@ -40,8 +40,8 @@ jobs: with: runner: linux.aws.h100 python-version: "3.12" - image: nvidia/cuda:12.9.1-devel-ubuntu24.04 - runtime-version: cu129 + image: nvidia/cuda:12.8.1-devel-ubuntu24.04 + runtime-version: cu128 container-options: --gpus all alias: h100 num-shards: ${{ matrix.num_shards }} @@ -65,8 +65,8 @@ jobs: with: runner: linux.dgx.b200 python-version: "3.12" - image: nvidia/cuda:12.9.1-devel-ubuntu24.04 - runtime-version: cu129 + image: nvidia/cuda:13.0.1-devel-ubuntu24.04 + runtime-version: cu130 container-options: --gpus all alias: b200 num-shards: ${{ matrix.num_shards }}