@@ -53,46 +53,6 @@ typedef enum wasi_nn_target {
5353 wasi_nn_unsupported_target ,
5454} wasi_nn_target ;
5555
56- static wasi_nn_encoding
57- str2encoding (char * str_encoding )
58- {
59- if (!str_encoding ) {
60- printf ("Got empty string encoding" );
61- return -1 ;
62- }
63-
64- if (!strcmp (str_encoding , "openvino" ))
65- return wasi_nn_openvino ;
66- else if (!strcmp (str_encoding , "tensorflowlite" ))
67- return wasi_nn_tensorflowlite ;
68- else if (!strcmp (str_encoding , "ggml" ))
69- return wasi_nn_ggml ;
70- else if (!strcmp (str_encoding , "onnx" ))
71- return wasi_nn_onnx ;
72- else
73- return wasi_nn_unknown_backend ;
74- // return autodetect;
75- }
76-
77- static wasi_nn_target
78- str2target (char * str_target )
79- {
80- if (!str_target ) {
81- printf ("Got empty string target" );
82- return -1 ;
83- }
84-
85- if (!strcmp (str_target , "cpu" ))
86- return wasi_nn_cpu ;
87- else if (!strcmp (str_target , "gpu" ))
88- return wasi_nn_gpu ;
89- else if (!strcmp (str_target , "tpu" ))
90- return wasi_nn_tpu ;
91- else
92- return wasi_nn_unsupported_target ;
93- // return autodetect;
94- }
95-
9656static void
9757libc_wasi_print_help (void )
9858{
@@ -245,6 +205,46 @@ libc_wasi_set_init_args(struct InstantiationArgs2 *args, int argc, char **argv,
245205}
246206
247207#if WASM_ENABLE_WASI_NN != 0 || WASM_ENABLE_WASI_EPHEMERAL_NN != 0
208+ static wasi_nn_encoding
209+ str2encoding (char * str_encoding )
210+ {
211+ if (!str_encoding ) {
212+ printf ("Got empty string encoding" );
213+ return -1 ;
214+ }
215+
216+ if (!strcmp (str_encoding , "openvino" ))
217+ return wasi_nn_openvino ;
218+ else if (!strcmp (str_encoding , "tensorflowlite" ))
219+ return wasi_nn_tensorflowlite ;
220+ else if (!strcmp (str_encoding , "ggml" ))
221+ return wasi_nn_ggml ;
222+ else if (!strcmp (str_encoding , "onnx" ))
223+ return wasi_nn_onnx ;
224+ else
225+ return wasi_nn_unknown_backend ;
226+ // return autodetect;
227+ }
228+
229+ static wasi_nn_target
230+ str2target (char * str_target )
231+ {
232+ if (!str_target ) {
233+ printf ("Got empty string target" );
234+ return -1 ;
235+ }
236+
237+ if (!strcmp (str_target , "cpu" ))
238+ return wasi_nn_cpu ;
239+ else if (!strcmp (str_target , "gpu" ))
240+ return wasi_nn_gpu ;
241+ else if (!strcmp (str_target , "tpu" ))
242+ return wasi_nn_tpu ;
243+ else
244+ return wasi_nn_unsupported_target ;
245+ // return autodetect;
246+ }
247+
248248libc_wasi_parse_result_t
249249wasi_nn_parse (char * * argv , wasi_nn_parse_context_t * ctx )
250250{
0 commit comments