bench : warm-up all kernels (#3438)
This commit is contained in:
parent
22c12ee86d
commit
a77d11d91e
|
|
@ -99,7 +99,15 @@ static int whisper_bench_full(const whisper_params & params) {
|
||||||
}
|
}
|
||||||
|
|
||||||
// text-generation heat
|
// text-generation heat
|
||||||
if (int ret = whisper_decode(ctx, tokens, 1, 256, params.n_threads) != 0) {
|
for (int i = 0; i < 256; i++) {
|
||||||
|
if (int ret = whisper_decode(ctx, tokens, 1, i, params.n_threads) != 0) {
|
||||||
|
fprintf(stderr, "error: failed to decode: %d\n", ret);
|
||||||
|
return 4;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// batched heat
|
||||||
|
if (int ret = whisper_decode(ctx, tokens, 5, 0, params.n_threads) != 0) {
|
||||||
fprintf(stderr, "error: failed to decode: %d\n", ret);
|
fprintf(stderr, "error: failed to decode: %d\n", ret);
|
||||||
return 4;
|
return 4;
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue