Browse Source
Merge pull request #5710 from dhiltgen/rocm_bump
Bump linux ROCm to 6.1.2
royh-openai-suffixdocs
Daniel Hiltgen
2 years ago
committed by
GitHub
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with
2 additions and
2 deletions
-
.github/workflows/test.yaml
-
Dockerfile
|
|
|
@ -126,7 +126,7 @@ jobs: |
|
|
|
strategy: |
|
|
|
matrix: |
|
|
|
rocm-version: |
|
|
|
- '6.1.1' |
|
|
|
- '6.1.2' |
|
|
|
runs-on: linux |
|
|
|
container: rocm/dev-ubuntu-20.04:${{ matrix.rocm-version }} |
|
|
|
steps: |
|
|
|
|
|
|
|
@ -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.1.1 |
|
|
|
ARG ROCM_VERSION=6.1.2 |
|
|
|
|
|
|
|
# Copy the minimal context we need to run the generate scripts |
|
|
|
FROM scratch AS llm-code |
|
|
|
|