chore: fix amd rocm build (#571)

This commit is contained in:
null-define
2025-01-18 13:11:39 +08:00
committed by GitHub
parent 27edb765a5
commit b70aaa672a

View File

@@ -54,7 +54,7 @@ endif ()
if (SD_HIPBLAS) if (SD_HIPBLAS)
message("-- Use HIPBLAS as backend stable-diffusion") message("-- Use HIPBLAS as backend stable-diffusion")
set(GGML_HIPBLAS ON) set(GGML_HIP ON)
add_definitions(-DSD_USE_CUDA) add_definitions(-DSD_USE_CUDA)
if(SD_FAST_SOFTMAX) if(SD_FAST_SOFTMAX)
set(GGML_CUDA_FAST_SOFTMAX ON) set(GGML_CUDA_FAST_SOFTMAX ON)