chore: fix amd rocm build (#571)
This commit is contained in:
parent
27edb765a5
commit
b70aaa672a
@ -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)
|
||||||
|
Loading…
Reference in New Issue
Block a user