Add CUDA support for Linux

build_cpp_linux.sh

@@ -11,15 +11,15 @@ clean_build(){
   cd "$build_path"
 }
 
-build_linux() {
+build_cpu() {
   clean_build
-  echo "Starting building for Linux..."
+  echo "Starting building for CPU..."
 
   cmake -DCMAKE_BUILD_TYPE=Release \
   -DWHISPER_BUILD_TESTS=OFF -DWHISPER_BUILD_EXAMPLES=OFF ../
   make
 
-  echo "Build for Linux complete!"
+  echo "Build for CPU complete!"
 
   artifact_path="$build_path/libwhisper.so"
   target_path="$unity_project/Packages/com.whisper.unity/Plugins/Linux/libwhisper.so"
@@ -28,4 +28,30 @@ build_linux() {
   echo "Build files copied to $target_path"
 }
 
-build_linux
+build_cuda() {
+  clean_build
+  echo "Starting building for CUDA..."
+
+  cmake -DWHISPER_CUBLAS=ON -DCMAKE_BUILD_TYPE=Release \
+  -DWHISPER_BUILD_TESTS=OFF -DWHISPER_BUILD_EXAMPLES=OFF ../
+  make
+
+  echo "Build for CUDA complete!"
+
+  artifact_path="$build_path/libwhisper.so"
+  target_path="$unity_project/Packages/com.whisper.unity/Plugins/Linux/libwhisper_cuda.so"
+  cp "$artifact_path" "$target_path"
+
+  echo "Build files copied to $target_path"
+}
+
+if [ "$targets" = "all" ]; then
+  build_cpu
+  build_cuda
+elif [ "$targets" = "cpu" ]; then
+  build_cpu
+elif [ "$targets" = "cuda" ]; then
+  build_cuda
+else
+  echo "Unknown targets: $targets"
+fi