@@ -10,7 +10,7 @@ option(WHISPERCPP_WITH_HIPBLAS "Build Whisper with hipBLAS support" OFF)
10
10
11
11
set (CMAKE_OSX_ARCHITECTURES_ "$ENV{MACOS_ARCH} " )
12
12
13
- set (Whispercpp_Build_GIT_TAG "f68298ce06ca3edd6e6f3f21c3d0bb5f073942c3 " )
13
+ set (Whispercpp_Build_GIT_TAG "c4e1861d2c24b186cbbac6c07480aaa298b0e6d9 " )
14
14
15
15
if (${CMAKE_BUILD_TYPE} STREQUAL Release OR ${CMAKE_BUILD_TYPE} STREQUAL
16
16
RelWithDebInfo)
@@ -41,18 +41,9 @@ endif()
41
41
42
42
if (WIN32 )
43
43
if (WHISPERCPP_WITH_CUDA)
44
- # Build with CUDA Check that CUDA_TOOLKIT_ROOT_DIR is set
45
- if (NOT DEFINED CUDA_TOOLKIT_ROOT_DIR)
46
- message (
47
- FATAL_ERROR
48
- "CUDA_TOOLKIT_ROOT_DIR is not set. Please set it to the root directory of your CUDA "
49
- "installation, e.g. `C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.4`"
50
- )
51
- endif (NOT DEFINED CUDA_TOOLKIT_ROOT_DIR)
52
-
53
- set (WHISPER_ADDITIONAL_ENV "CUDAToolkit_ROOT=${CUDA_TOOLKIT_ROOT_DIR} " )
54
- set (WHISPER_ADDITIONAL_CMAKE_ARGS
55
- -DGGML_CUDA=ON -DCMAKE_GENERATOR_TOOLSET=cuda=${CUDA_TOOLKIT_ROOT_DIR} )
44
+ # Build with CUDA
45
+ set (WHISPER_ADDITIONAL_ENV "" )
46
+ set (WHISPER_ADDITIONAL_CMAKE_ARGS -DGGML_CUDA=ON )
56
47
set (WHISPER_CMAKE_GENERATOR ${CMAKE_GENERATOR} )
57
48
elseif (WHISPERCPP_WITH_HIPBLAS)
58
49
# Build with hipBLAS
@@ -104,12 +95,8 @@ if(WIN32)
104
95
<INSTALL_DIR>/bin/${CMAKE_SHARED_LIBRARY_PREFIX} whisper${CMAKE_SHARED_LIBRARY_SUFFIX}
105
96
<INSTALL_DIR>/lib/${CMAKE_IMPORT_LIBRARY_PREFIX} whisper${CMAKE_IMPORT_LIBRARY_SUFFIX}
106
97
CMAKE_GENERATOR ${CMAKE_GENERATOR}
107
- INSTALL_COMMAND
108
- ${CMAKE_COMMAND} --install <BINARY_DIR> --config ${Whispercpp_BUILD_TYPE}
109
- && ${CMAKE_COMMAND} -E copy
110
- <BINARY_DIR>/${Whispercpp_BUILD_TYPE} /whisper.lib <INSTALL_DIR>/lib &&
111
- ${CMAKE_COMMAND} -E copy <SOURCE_DIR>/ggml/include /ggml.h
112
- <INSTALL_DIR>/include
98
+ INSTALL_COMMAND ${CMAKE_COMMAND} --install <BINARY_DIR> --config
99
+ ${Whispercpp_BUILD_TYPE}
113
100
CONFIGURE_COMMAND
114
101
${CMAKE_COMMAND} -E env ${WHISPER_ADDITIONAL_ENV} ${CMAKE_COMMAND}
115
102
<SOURCE_DIR> -B <BINARY_DIR> -G ${WHISPER_CMAKE_GENERATOR}
@@ -179,6 +166,15 @@ if(WIN32)
179
166
"${HIP_PATH_STR} /bin/amd_comgr_2.dll" )
180
167
install (FILES ${HIPBLAS_DLLS} DESTINATION ${CMAKE_SOURCE_DIR} /release/bin)
181
168
else ()
169
+ # Check that CUDA_TOOLKIT_ROOT_DIR is set
170
+ if (NOT DEFINED CUDA_TOOLKIT_ROOT_DIR)
171
+ message (
172
+ FATAL_ERROR
173
+ "CUDA_TOOLKIT_ROOT_DIR is not set. Please set it to the root directory of your CUDA "
174
+ "installation, e.g. `C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.4`"
175
+ )
176
+ endif (NOT DEFINED CUDA_TOOLKIT_ROOT_DIR)
177
+
182
178
# normalize CUDA path with file(TO_CMAKE_PATH)
183
179
file (TO_CMAKE_PATH ${CUDA_TOOLKIT_ROOT_DIR} CUDA_TOOLKIT_ROOT_DIR)
184
180
# find the CUDA DLLs for cuBLAS in the bin directory of the CUDA
0 commit comments