소스 검색

nn-cli: fix get_output index (#4543)

YAMAMOTO Takashi 5 달 전
부모
커밋
f90cc086ad
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      wamr-wasi-extensions/samples/nn-cli/main.c

+ 1 - 1
wamr-wasi-extensions/samples/nn-cli/main.c

@@ -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;