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); exit(1);
} }
nnret = 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) { if (nnret == wasi_ephemeral_nn_error_too_large) {
resultbufsz *= 2; resultbufsz *= 2;
goto retry; goto retry;