whisper : enable IGPU (#3492)

Co-authored-by: Oleg Orlov <vk.orelsokolov@yandex.by>
This commit is contained in:
Oleg Orlov 2025-11-01 15:38:28 +03:00 committed by GitHub
parent c62adfbd1e
commit 999a7e0cbf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 1 deletions

View File

@ -1296,7 +1296,11 @@ static ggml_backend_t whisper_backend_init_gpu(const whisper_context_params & pa
if (params.use_gpu) {
for (size_t i = 0; i < ggml_backend_dev_count(); ++i) {
ggml_backend_dev_t dev_cur = ggml_backend_dev_get(i);
if (ggml_backend_dev_type(dev_cur) == GGML_BACKEND_DEVICE_TYPE_GPU || ggml_backend_dev_type(dev_cur) == GGML_BACKEND_DEVICE_TYPE_IGPU) {
enum ggml_backend_dev_type dev_type = ggml_backend_dev_type(dev_cur);
const char * dev_name = ggml_backend_dev_name(dev_cur);
WHISPER_LOG_INFO("%s: device %zu: %s (type: %d)\n", __func__, i, dev_name, dev_type);
if (dev_type == GGML_BACKEND_DEVICE_TYPE_GPU || dev_type == GGML_BACKEND_DEVICE_TYPE_IGPU) {
WHISPER_LOG_INFO("%s: found GPU device %zu: %s (type: %d, cnt: %d)\n", __func__, i, dev_name, dev_type, cnt);
if (cnt == params.gpu_device) {
dev = dev_cur;
}