feat: add support for setInput in EngineHailo

Seeed-Studio · Nov 28, 2024 · 9bf406c · 9bf406c
1 parent 9053a4c
commit 9bf406c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/sscma/core/engine/ma_engine_hailo.cpp b/sscma/core/engine/ma_engine_hailo.cpp
@@ -396,6 +396,16 @@ ma_quant_param_t EngineHailo::getOutputQuantParam(int32_t index) {
 
 
 ma_err_t EngineHailo::setInput(int32_t index, const ma_tensor_t& tensor) {
+    if (index < 0 || index >= static_cast<int32_t>(_input_tensors.size())) {
+        return MA_EINVAL;
+    }
+
+    if (tensor.size != _input_tensors[index]->size) {
+        return MA_EINVAL;
+    }
+
+    std::memcpy(_input_tensors[index]->data.data, tensor.data.data, tensor.size);
+
     return MA_ENOTSUP;
 }