ggerganov commited on
Commit
f425556
·
unverified ·
1 Parent(s): d321914

whisper : reset conv scheduler when CoreML is used (#3350)

Browse files
Files changed (1) hide show
  1. src/whisper.cpp +2 -0
src/whisper.cpp CHANGED
@@ -2432,6 +2432,8 @@ static bool whisper_encode_internal(
2432
  return false;
2433
  }
2434
  } else {
 
 
2435
  #if defined(WHISPER_USE_COREML)
2436
  whisper_coreml_encode(wstate.ctx_coreml, mel->ne[0], mel->ne[1], (float *) mel->data, (float *) wstate.embd_enc->data);
2437
  #elif defined(WHISPER_USE_OPENVINO)
 
2432
  return false;
2433
  }
2434
  } else {
2435
+ ggml_backend_sched_reset(sched);
2436
+
2437
  #if defined(WHISPER_USE_COREML)
2438
  whisper_coreml_encode(wstate.ctx_coreml, mel->ne[0], mel->ne[1], (float *) mel->data, (float *) wstate.embd_enc->data);
2439
  #elif defined(WHISPER_USE_OPENVINO)