Skip to content

Commit 2103ab6

Browse files
committed
fixed rocm detection by adding gfx targets
Signed-off-by: Brian <[email protected]>
1 parent 42b88e3 commit 2103ab6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

container-images/scripts/build_llama_and_whisper.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ configure_common_flags() {
6262
common_flags=("-DGGML_NATIVE=OFF")
6363
case "$containerfile" in
6464
rocm)
65-
common_flags+=("-DGGML_HIP=ON")
65+
common_flags+=("-DGGML_HIP=ON" "-DAMDGPU_TARGETS=gfx1010,gfx1030,gfx1032,gfx1100,gfx1101,gfx1102")
6666
;;
6767
cuda)
6868
common_flags+=("-DGGML_CUDA=ON" "-DCMAKE_EXE_LINKER_FLAGS=-Wl,--allow-shlib-undefined")

0 commit comments

Comments
 (0)