Skip to content

Commit 5527454

Browse files
authored
whisper : fixed crash in GPU device selection on multi-GPU systems (#3372)
1 parent b02242d commit 5527454

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/whisper.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1327,7 +1327,7 @@ static ggml_backend_t whisper_backend_init_gpu(const whisper_context_params & pa
13271327
for (size_t i = 0; i < ggml_backend_dev_count(); ++i) {
13281328
ggml_backend_dev_t dev_cur = ggml_backend_dev_get(i);
13291329
if (ggml_backend_dev_type(dev_cur) == GGML_BACKEND_DEVICE_TYPE_GPU) {
1330-
if (cnt == 0 || cnt == params.gpu_device) {
1330+
if (cnt == params.gpu_device) {
13311331
dev = dev_cur;
13321332
}
13331333

@@ -1396,7 +1396,7 @@ static buft_list_t make_buft_list(whisper_context_params & params) {
13961396
for (size_t i = 0; i < ggml_backend_dev_count(); ++i) {
13971397
ggml_backend_dev_t dev = ggml_backend_dev_get(i);
13981398
if (ggml_backend_dev_type(dev) == GGML_BACKEND_DEVICE_TYPE_GPU) {
1399-
if (cnt == 0 || cnt == params.gpu_device) {
1399+
if (cnt == params.gpu_device) {
14001400
auto * buft = ggml_backend_dev_buffer_type(dev);
14011401
if (buft) {
14021402
buft_list.emplace_back(dev, buft);

0 commit comments

Comments
 (0)