whisper : enable IGPU (#3492)
Co-authored-by: Oleg Orlov <vk.orelsokolov@yandex.by>
This commit is contained in:
parent
c62adfbd1e
commit
999a7e0cbf
|
|
@ -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;
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue