add stable-diffusion-abi feature for golang

Author: Cyberhan123

.gitignore

@@ -0,0 +1 @@
+/build/

.idea/.gitignore

@@ -0,0 +1,98 @@
+cmake_minimum_required(VERSION 3.12)
+project("stable-diffusion-build")
+
+set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
+
+if (NOT XCODE AND NOT MSVC AND NOT CMAKE_BUILD_TYPE)
+    set(CMAKE_BUILD_TYPE Release CACHE STRING "Build type" FORCE)
+    set_property(CACHE CMAKE_BUILD_TYPE PROPERTY STRINGS "Debug" "Release" "MinSizeRel" "RelWithDebInfo")
+endif()
+
+set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/bin)
+set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/bin)
+
+set(CMAKE_CXX_STANDARD 11)
+set(CMAKE_CXX_STANDARD_REQUIRED ON)
+set(CMAKE_C_STANDARD 11)
+set(CMAKE_C_STANDARD_REQUIRED true)
+set(THREADS_PREFER_PTHREAD_FLAG ON)
+find_package(Threads REQUIRED)
+
+set(CMAKE_MODULE_PATH "${CMAKE_SOURCE_DIR}/cmake;${CMAKE_MODULE_PATH}")
+
+# Instruction set specific
+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)
+
+
+
+if (MSVC)
+    add_compile_definitions(_CRT_SECURE_NO_WARNINGS)
+
+    if (BUILD_SHARED_LIBS)
+        set(CMAKE_WINDOWS_EXPORT_ALL_SYMBOLS ON)
+    endif ()
+endif ()
+
+if (${CMAKE_SYSTEM_PROCESSOR} MATCHES "arm" OR ${CMAKE_SYSTEM_PROCESSOR} MATCHES "aarch64")
+    message(STATUS "ARM detected")
+    if (MSVC)
+        # TODO [llama.cpp]: arm msvc?
+    else()
+        if (${CMAKE_SYSTEM_PROCESSOR} MATCHES "aarch64")
+            add_compile_options(-mcpu=native)
+        endif()
+        # TODO [llama.cpp]: armv6,7,8 version specific flags
+    endif()
+elseif (${CMAKE_SYSTEM_PROCESSOR} MATCHES "^(x86_64|i686|AMD64)$")
+    message(STATUS "x86 detected")
+    if (MSVC)
+        if (SD_AVX512)
+            add_compile_options($<$<COMPILE_LANGUAGE:C>:/arch:AVX512>)
+            add_compile_options($<$<COMPILE_LANGUAGE:CXX>:/arch:AVX512>)
+            # MSVC has no compile-time flags enabling specific
+            # AVX512 extensions, neither it defines the
+            # macros corresponding to the extensions.
+            # Do it manually.
+        elseif (SD_AVX2)
+            add_compile_options($<$<COMPILE_LANGUAGE:C>:/arch:AVX2>)
+            add_compile_options($<$<COMPILE_LANGUAGE:CXX>:/arch:AVX2>)
+        elseif (SD_AVX)
+            add_compile_options($<$<COMPILE_LANGUAGE:C>:/arch:AVX>)
+            add_compile_options($<$<COMPILE_LANGUAGE:CXX>:/arch:AVX>)
+        endif()
+    else()
+        add_compile_options(-mf16c)
+        if (SD_FMA)
+            add_compile_options(-mfma)
+        endif()
+        if (SD_AVX)
+            add_compile_options(-mavx)
+        endif()
+        if (SD_AVX2)
+            add_compile_options(-mavx2)
+        endif()
+        if (SD_AVX512)
+            add_compile_options(-mavx512f)
+            add_compile_options(-mavx512bw)
+        endif()
+    endif()
+else()
+    # TODO [llama.cpp]: support PowerPC
+    message(STATUS "Unknown architecture")
+endif()
+# deps
+
+include(sd)
+include_directories(${sd_SOURCE_DIR})
+
+set(SD_ABI sd-abi)
+set(SD_EXTRA_LIBS "")
+add_library(${SD_ABI} SHARED stable-diffusion-abi.cpp stable-diffusion-abi.h)
+set_target_properties(${SD_ABI} PROPERTIES POSITION_INDEPENDENT_CODE ON)
+target_compile_definitions(${SD_ABI} PRIVATE STABLE_DIFFUSION_SHARED STABLE_DIFFUSION_BUILD)
+target_link_libraries(${SD_ABI} PRIVATE $<TARGET_OBJECTS:ggml> $<TARGET_OBJECTS:stable-diffusion> ${SD_EXTRA_LIBS} ${CMAKE_THREAD_LIBS_INIT})
+target_include_directories(${SD_ABI} PUBLIC .)
+

.idea/modules.xml

@@ -1 +1,11 @@
-# stable-diffusion.cpp-build
+# stable-diffusion.cpp-build
+
+
+## Build
+
+```commandline
+mkdir build
+cd build
+cmake .. -DCMAKE_BUILD_TYPE=Release 
+cmake --build . --config Release
+```

.idea/stable-diffusion.cpp-build.iml

@@ -0,0 +1,14 @@
+# Campatible with cmake 3.11 and above.
+macro(FetchContent_MakeAvailable NAME)
+FetchContent_GetProperties(${NAME})
+if(NOT ${NAME}_POPULATED)
+    FetchContent_Populate(${NAME})
+    add_subdirectory(${${NAME}_SOURCE_DIR} ${${NAME}_BINARY_DIR})
+endif()
+endmacro()macro(FetchContent_MakeAvailable NAME)
+    FetchContent_GetProperties(${NAME})
+    if(NOT ${NAME}_POPULATED)
+        FetchContent_Populate(${NAME})
+        add_subdirectory(${${NAME}_SOURCE_DIR} ${${NAME}_BINARY_DIR})
+    endif()
+endmacro()

.idea/vcs.xml

@@ -0,0 +1,16 @@
+include(FetchContent)
+# FetchContent_MakeAvailable was not added until CMake 3.14
+if(${CMAKE_VERSION} VERSION_LESS 3.14)
+    include(add_FetchContent_MakeAvailable.cmake)
+endif()
+
+set(SD_GIT_TAG  fb7d9217bb5dd6ac82b4fb1f312381f63b65ba1a)
+set(SD_GIT_URL  https://github.com/Cyberhan123/stable-diffusion.cpp)
+
+FetchContent_Declare(
+  sd
+  GIT_REPOSITORY    ${SD_GIT_URL}
+  GIT_TAG           ${SD_GIT_TAG}
+)
+
+FetchContent_MakeAvailable(sd)