diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml index e0761ccd..d5d9b7ea 100644 --- a/.github/workflows/test.yaml +++ b/.github/workflows/test.yaml @@ -106,7 +106,7 @@ jobs: strategy: matrix: rocm-version: - - '6.0' + - '6.0.2' runs-on: linux container: rocm/dev-ubuntu-20.04:${{ matrix.rocm-version }} steps: diff --git a/Dockerfile b/Dockerfile index 919d48de..41081ce7 100644 --- a/Dockerfile +++ b/Dockerfile @@ -2,7 +2,7 @@ ARG GOLANG_VERSION=1.22.1 ARG CMAKE_VERSION=3.22.1 # this CUDA_VERSION corresponds with the one specified in docs/gpu.md ARG CUDA_VERSION=11.3.1 -ARG ROCM_VERSION=6.0 +ARG ROCM_VERSION=6.0.2 # Copy the minimal context we need to run the generate scripts FROM scratch AS llm-code