parent
a95631ee97
commit
0e814dfc42
12 changed files with 19 additions and 103 deletions
|
@ -36,7 +36,7 @@ COPY . .
|
|||
# Set nvcc architecture
|
||||
ENV GPU_TARGETS=${ROCM_DOCKER_ARCH}
|
||||
# Enable ROCm
|
||||
ENV LLAMA_HIPBLAS=1
|
||||
ENV GGML_HIPBLAS=1
|
||||
ENV CC=/opt/rocm/llvm/bin/clang
|
||||
ENV CXX=/opt/rocm/llvm/bin/clang++
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue