cublas

Author: ring-c

.gitignore

@@ -5,3 +5,4 @@
 /cmake-build-release/
 /cmake-build-debug-visual-studio/
 /cmake-build-release-visual-studio/
+/.idea

.idea/.name

@@ -29,6 +29,7 @@ option(SD_AVX "sd: enable AVX" ON)
 option(SD_AVX2 "sd: enable AVX2" ON)
 option(SD_AVX512 "sd: enable AVX512" OFF)
 option(SD_FMA "sd: enable FMA" ON)
+option(SD_CUBLAS "sd: cuda backend" OFF)
 
 if (SD_AVX512)
     set(GGML_AVX512 ON)
@@ -41,6 +42,12 @@ if (SD_FMA)
     set(GGML_FMA ON)
 endif ()
 
+if(SD_CUBLAS)
+	message("Use CUBLAS as backend stable-diffusion")
+    set(GGML_CUBLAS ON)
+    add_definitions(-DSD_USE_CUBLAS)
+endif()
+
 include(sd)
 include_directories(${sd_SOURCE_DIR})