From 8743f8cb13ac8538afb5d48763dab97094236b35 Mon Sep 17 00:00:00 2001
From: Ruwen Hahn <haruwenz@twitch.tv>
Date: Tue, 6 Aug 2024 16:43:11 +0200
Subject: [PATCH] Store acceleration info in cmake cache

This is to allow switching branches and rebuilding with the "same"
settings from e.g. visual studio (which will re-run a bunch of CMake
processing)
---
 CMakeLists.txt               | 10 ++++++++++
 cmake/BuildCTranslate2.cmake |  6 +++---
 cmake/BuildWhispercpp.cmake  | 18 +++++++++---------
 3 files changed, 22 insertions(+), 12 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 108ff06..91e54e0 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -48,6 +48,16 @@ else()
   target_link_libraries(${CMAKE_PROJECT_NAME} PRIVATE libcurl)
 endif()
 
+if(WIN32)
+  if(DEFINED ENV{ACCELERATION})
+    set(ACCELERATION $ENV{ACCELERATION} CACHE STRING "Acceleration to use" FORCE)
+  ENDIF()
+  if(NOT DEFINED ACCELERATION)
+    set(ACCELERATION "cpu" CACHE STRING "Acceleration to use")
+  endif()
+  set_property(CACHE ACCELERATION PROPERTY STRINGS "cpu" "hipblas" "cuda")
+endif()
+
 include(cmake/BuildWhispercpp.cmake)
 target_link_libraries(${CMAKE_PROJECT_NAME} PRIVATE Whispercpp)
 
diff --git a/cmake/BuildCTranslate2.cmake b/cmake/BuildCTranslate2.cmake
index 042fa89..0d60561 100644
--- a/cmake/BuildCTranslate2.cmake
+++ b/cmake/BuildCTranslate2.cmake
@@ -20,11 +20,11 @@ if(APPLE)
 elseif(WIN32)
 
   # check ACCELERATION environment variable
-  if(NOT DEFINED ENV{ACCELERATION})
-    message(FATAL_ERROR "Please set the ACCELERATION environment variable to either `cpu`, `hipblas`, or `cuda`")
+  if(NOT DEFINED ACCELERATION)
+    message(FATAL_ERROR "Please set ACCELERATION to either `cpu`, `hipblas`, or `cuda`")
   endif()
 
-  if($ENV{ACCELERATION} STREQUAL "cpu" OR $ENV{ACCELERATION} STREQUAL "hipblas")
+  if(${ACCELERATION} STREQUAL "cpu" OR ${ACCELERATION} STREQUAL "hipblas")
     FetchContent_Declare(
       ctranslate2_fetch
       URL https://github.com/occ-ai/obs-ai-ctranslate2-dep/releases/download/1.2.0/libctranslate2-windows-4.1.1-Release-cpu.zip
diff --git a/cmake/BuildWhispercpp.cmake b/cmake/BuildWhispercpp.cmake
index 18c76da..4f85cdd 100644
--- a/cmake/BuildWhispercpp.cmake
+++ b/cmake/BuildWhispercpp.cmake
@@ -46,22 +46,22 @@ if(APPLE)
       ${whispercpp_fetch_SOURCE_DIR}/lib/${CMAKE_STATIC_LIBRARY_PREFIX}whisper.coreml${CMAKE_STATIC_LIBRARY_SUFFIX})
 
 elseif(WIN32)
-  if(NOT DEFINED ENV{ACCELERATION})
+  if(NOT DEFINED ACCELERATION)
     message(
-      FATAL_ERROR "The ACCELERATION environment variable is not set. Please set it to either `cpu`, `cuda` or `hipblas`"
+      FATAL_ERROR "ACCELERATION is not set. Please set it to either `cpu`, `cuda` or `hipblas`"
     )
-  endif(NOT DEFINED ENV{ACCELERATION})
+  endif()
 
-  set(ARCH_PREFIX $ENV{ACCELERATION})
+  set(ARCH_PREFIX ${ACCELERATION})
   set(WHISPER_CPP_URL
       "${PREBUILT_WHISPERCPP_URL_BASE}/whispercpp-windows-${ARCH_PREFIX}-${PREBUILT_WHISPERCPP_VERSION}.zip")
-  if($ENV{ACCELERATION} STREQUAL "cpu")
+  if(${ACCELERATION} STREQUAL "cpu")
     set(WHISPER_CPP_HASH "2b1cfa0dd764132c4cde60e112a8e6328d28d158d91a8845080baa3e9d2dcdcd")
     add_compile_definitions("LOCALVOCAL_WITH_CPU")
-  elseif($ENV{ACCELERATION} STREQUAL "cuda")
+  elseif(${ACCELERATION} STREQUAL "cuda")
     set(WHISPER_CPP_HASH "011e813742fddf0911c4a36d2080d7a388cf78738081297088e7d50023e4f9bc")
     add_compile_definitions("LOCALVOCAL_WITH_CUDA")
-  elseif($ENV{ACCELERATION} STREQUAL "hipblas")
+  elseif(${ACCELERATION} STREQUAL "hipblas")
     set(WHISPER_CPP_HASH "f2980d6cd3df9cac464378d26d2c19d827bcac995c8d0398a39230a9be936013")
     add_compile_definitions("LOCALVOCAL_WITH_HIPBLAS")
   else()
@@ -90,7 +90,7 @@ elseif(WIN32)
   set_target_properties(Whispercpp::Whisper PROPERTIES INTERFACE_INCLUDE_DIRECTORIES
                                                        ${whispercpp_fetch_SOURCE_DIR}/include)
 
-  if($ENV{ACCELERATION} STREQUAL "cpu")
+  if(${ACCELERATION} STREQUAL "cpu")
     # add openblas to the link line
     add_library(Whispercpp::OpenBLAS STATIC IMPORTED)
     set_target_properties(Whispercpp::OpenBLAS PROPERTIES IMPORTED_LOCATION
@@ -143,7 +143,7 @@ endif()
 add_library(Whispercpp INTERFACE)
 add_dependencies(Whispercpp Whispercpp_Build)
 target_link_libraries(Whispercpp INTERFACE Whispercpp::Whisper)
-if(WIN32 AND "$ENV{ACCELERATION}" STREQUAL "cpu")
+if(WIN32 AND "${ACCELERATION}" STREQUAL "cpu")
   target_link_libraries(Whispercpp INTERFACE Whispercpp::OpenBLAS)
 endif()
 if(APPLE)