add wasi_ephemeral_nn in header file

Author: HongxiaWangSSSS

core/iwasm/libraries/wasi-nn/include/wasi_nn.h

@@ -15,6 +15,44 @@
 #include <stdint.h>
 #include "wasi_nn_types.h"
 
+#if WASM_ENABLE_WASI_EPHEMERAL_NN != 0
+wasi_nn_error
+load(graph_builder *builder, uint32_t builder_wasm_size,
+     graph_encoding encoding, execution_target target, graph *g)
+    __attribute__((import_module("wasi_ephemeral_nn")));
+
+wasi_nn_error
+load_by_name(char *name, uint32_t name_len, graph *g)
+    __attribute__((import_module("wasi_ephemeral_nn")));
+
+wasi_nn_error
+load_by_name_with_config(const char *name, uint32_t name_len, void *config,
+                         uint32_t config_len, graph *g)
+    __attribute__((import_module("wasi_ephemeral_nn")));
+/**
+ * INFERENCE
+ *
+ */
+
+wasi_nn_error
+init_execution_context(graph g, graph_execution_context *exec_ctx)
+    __attribute__((import_module("wasi_ephemeral_nn")));
+
+wasi_nn_error
+set_input(graph_execution_context ctx, uint32_t index, tensor *tensor)
+    __attribute__((import_module("wasi_ephemeral_nn")));
+
+wasi_nn_error
+compute(graph_execution_context ctx)
+    __attribute__((import_module("wasi_ephemeral_nn")));
+
+wasi_nn_error
+get_output(graph_execution_context ctx, uint32_t index,
+           tensor_data output_tensor, uint32_t output_tensor_len,
+           uint32_t *output_tensor_size)
+    __attribute__((import_module("wasi_ephemeral_nn")));
+#else
+
 /**
  * @brief Load an opaque sequence of bytes to use for inference.
  *
@@ -86,5 +124,5 @@ wasi_nn_error
 get_output(graph_execution_context ctx, uint32_t index,
            tensor_data output_tensor, uint32_t *output_tensor_size)
     __attribute__((import_module("wasi_nn")));
-
+#endif
 #endif

core/iwasm/libraries/wasi-nn/src/wasi_nn.c

@@ -697,6 +697,7 @@ static NativeSymbol native_symbols_wasi_nn[] = {
     REG_NATIVE_FUNC(get_output, "(ii*i*)i"),
 #else  /* WASM_ENABLE_WASI_EPHEMERAL_NN == 0 */
     REG_NATIVE_FUNC(load, "(*ii*)i"),
+    REG_NATIVE_FUNC(load_by_name, "(*i*)i"),
     REG_NATIVE_FUNC(init_execution_context, "(i*)i"),
     REG_NATIVE_FUNC(set_input, "(ii*)i"),
     REG_NATIVE_FUNC(compute, "(i)i"),

core/iwasm/libraries/wasi-nn/test/utils.c

@@ -99,7 +99,11 @@ wasi_nn_error
 wasm_get_output(graph_execution_context ctx, uint32_t index, float *out_tensor,
                 uint32_t *out_size)
 {
+#if WASM_ENABLE_WASI_EPHEMERAL_NN != 0
+    return get_output(ctx, index, (uint8_t *)out_tensor, *out_size, out_size);
+#else
     return get_output(ctx, index, (uint8_t *)out_tensor, out_size);
+#endif
 }
 
 float *