瀏覽代碼

Merge pull request #5710 from dhiltgen/rocm_bump

Bump linux ROCm to 6.1.2
Daniel Hiltgen 9 月之前
父節點
當前提交
8288ec8824
共有 2 個文件被更改,包括 2 次插入2 次删除
  1. 1 1
      .github/workflows/test.yaml
  2. 1 1
      Dockerfile

+ 1 - 1
.github/workflows/test.yaml

@@ -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:

+ 1 - 1
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.1.1
+ARG ROCM_VERSION=6.1.2
 
 # Copy the minimal context we need to run the generate scripts
 FROM scratch AS llm-code