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)