nn-cli: fix get_output index (#4543)

This commit is contained in:
YAMAMOTO Takashi 2025-08-11 16:05:44 +09:00 committed by GitHub
parent 6b51c61f5e
commit f90cc086ad
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -406,7 +406,7 @@ retry:
exit(1);
}
nnret =
wasi_ephemeral_nn_get_output(c, 0, resultbuf, resultbufsz, &resultsz);
wasi_ephemeral_nn_get_output(c, idx, resultbuf, resultbufsz, &resultsz);
if (nnret == wasi_ephemeral_nn_error_too_large) {
resultbufsz *= 2;
goto retry;