extern/whisper.cpp
1
0
Fork 1
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2025-08-09 15:45:57 +02:00
Code Issues Packages Projects Releases Wiki Activity

whisper : expose CUDA device setting in public API (#1840)

Browse Source 
* Makefile : allow to override CUDA_ARCH_FLAG

* whisper : allow to select GPU (CUDA) device from public API
This commit is contained in:
Didzis Gosko
2024-02-09 17:27:47 +02:00
committed by GitHub
parent b6559333ff
commit 0f80e5a80a
3 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
whisper.h
View File

@ -86,6 +86,7 @@ extern "C" {
struct whisper_context_params {
bool use_gpu;
int gpu_device; // CUDA device
};
typedef struct whisper_token_data {