From dda3c8d03fb2f349f21072565c8c644b8fd5c52d Mon Sep 17 00:00:00 2001 From: Mathieu Broillet Date: Mon, 23 Sep 2024 18:28:48 +0200 Subject: [PATCH] update rocm to 6.2 --- .github/workflows/build-and_publish-prebuilts-whl.yml | 6 +++--- prebuilts/Dockerfile | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/build-and_publish-prebuilts-whl.yml b/.github/workflows/build-and_publish-prebuilts-whl.yml index 40433bd..d1d05e0 100644 --- a/.github/workflows/build-and_publish-prebuilts-whl.yml +++ b/.github/workflows/build-and_publish-prebuilts-whl.yml @@ -6,11 +6,11 @@ on: rocm_version: description: 'ROCm version' required: true - default: '6.1.2' + default: '6.2' torch_version: - description: 'Torch version (e.g., rocm6.1)' + description: 'Torch version (e.g., rocm6.2)' required: true - default: 'rocm6.1' + default: 'rocm6.2' jobs: build: diff --git a/prebuilts/Dockerfile b/prebuilts/Dockerfile index cb3e7aa..369afe8 100644 --- a/prebuilts/Dockerfile +++ b/prebuilts/Dockerfile @@ -1,4 +1,4 @@ -ARG ROCM_VERSION=6.1.2 +ARG ROCM_VERSION=6.2 FROM rocm/dev-ubuntu-22.04:${ROCM_VERSION} # also works without -complete for specific arch (tested on gfx1030)