[api] Add a new nnfw EXECUTORCH_LLAMA

- Add a enum for new nnfw `EXECUTORCH_LLAMA` Signed-off-by: Yongjoo Ahn <[email protected]>
nnstreamer · Oct 29, 2024 · 120c938 · 120c938
1 parent f5b00af
commit 120c938
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/c/include/ml-api-common.h b/c/include/ml-api-common.h
@@ -75,6 +75,7 @@ typedef enum {
   ML_NNFW_TYPE_NCNN = 18,             /**< Tencent ncnn (Since 9.0) */
   ML_NNFW_TYPE_TENSORRT = 19,         /**< NVidia Tensor-RT (Since 9.0) */
   ML_NNFW_TYPE_QNN = 20,              /**< Qualcomm QNN (Qualcomm® AI Engine Direct) (Since 9.0) */
+  ML_NNFW_TYPE_EXECUTORCH_LLAMA = 21, /**< ExecuTorch Llama runner */
   ML_NNFW_TYPE_SNAP = 0x2001,         /**< SNAP (Samsung Neural Acceleration Platform), only for Android. (Since 6.0) */
 } ml_nnfw_type_e;
 

diff --git a/c/src/ml-api-inference-single.c b/c/src/ml-api-inference-single.c
@@ -112,6 +112,7 @@ static const char *ml_nnfw_subplugin_name[] = {
   [ML_NNFW_TYPE_NCNN] = "ncnn",
   [ML_NNFW_TYPE_TENSORRT] = "tensorrt",
   [ML_NNFW_TYPE_QNN] = "qnn",
+  [ML_NNFW_TYPE_EXECUTORCH_LLAMA] = "executorch-llama",
   NULL
 };
 
@@ -1957,6 +1958,7 @@ _ml_validate_model_file (const char *const *model,
     case ML_NNFW_TYPE_ONNX_RUNTIME:
     case ML_NNFW_TYPE_NCNN:
     case ML_NNFW_TYPE_TENSORRT:
+    case ML_NNFW_TYPE_EXECUTORCH_LLAMA:
     case ML_NNFW_TYPE_QNN:
       /**
        * We cannot check the file ext with NNFW.