triggerflow 0.2.5__py3-none-any.whl → 0.2.7__py3-none-any.whl

triggerflow/templates/model_template.cpp

@@ -2,8 +2,6 @@
 #include "emulator.h"
 #include "NN/nnet_utils/nnet_common.h"
 #include <any>
-#include <array>
-#include <utility>
 #include "ap_fixed.h"
 #include "ap_int.h"
 #include "scales.h"
@@ -14,40 +12,45 @@ class {{MODEL_NAME}}_emulator : public hls4mlEmulator::Model {
 
 private:
     typedef {{UNSCALED_TYPE}} unscaled_t;
-    static const int N_INPUT_SIZE = {{N_INPUTS}};
+    static const int N_INPUT_SIZE  = {{N_INPUTS}};
+    static const int N_OUTPUT_SIZE = {{N_OUTPUTS}};
 
     unscaled_t _unscaled_input[N_INPUT_SIZE];
-    {{MODEL_NAME}}::input_t _scaled_input;
-    {{MODEL_NAME}}::result_t _result;
+    {{MODEL_NAME}}::input_t  _scaled_input[N_INPUT_SIZE];
+    {{MODEL_NAME}}::result_t _result[N_OUTPUT_SIZE];
 
-    // Scale the raw input array to the model input type
-    virtual void _scaleNNInputs(unscaled_t unscaled[N_INPUT_SIZE], {{MODEL_NAME}}::input_t &scaled)
+    virtual void _scaleNNInputs(unscaled_t unscaled[N_INPUT_SIZE],
+                                {{MODEL_NAME}}::input_t scaled[N_INPUT_SIZE])
     {
-        for (int i = 0; i < N_INPUT_SIZE; i++)
-        {
+        for (int i = 0; i < N_INPUT_SIZE; i++) {
             unscaled_t tmp0 = unscaled[i] - hls4ml_{{MODEL_NAME}}::ad_offsets[i];
-            {{UNSCALED_TYPE}} tmp1 = tmp0 >> hls4ml_{{MODEL_NAME}}::ad_shift[i];
+            {{MODEL_NAME}}::input_t tmp1 = tmp0 >> hls4ml_{{MODEL_NAME}}::ad_shift[i];
             scaled[i] = tmp1;
         }
     }
 
 public:
-    virtual void prepare_input(std::any input) {
-        unscaled_t *unscaled_input_p = std::any_cast<unscaled_t*>(input);
+    virtual void prepare_input(std::any input) override {
+        unscaled_t* unscaled_input_p = std::any_cast<unscaled_t*>(input);
+
         for (int i = 0; i < N_INPUT_SIZE; i++) {
             _unscaled_input[i] = unscaled_input_p[i];
         }
+
         _scaleNNInputs(_unscaled_input, _scaled_input);
     }
 
-    virtual void predict() {
-        // Call the io_parallel model function; pass pointers
-        {{MODEL_NAME}}::{{MODEL_NAME}}_project(&_scaled_input, &_result);
+    virtual void predict() override {
+        {{MODEL_NAME}}::{{MODEL_NAME}}_project(_scaled_input, _result);
     }
 
-    virtual void read_result(std::any result) {
-        {{MODEL_NAME}}::result_t *result_p = std::any_cast<{{MODEL_NAME}}::result_t*>(result);
-        *result_p = _result;
+    virtual void read_result(std::any result) override {
+        {{MODEL_NAME}}::result_t* result_p =
+            std::any_cast<{{MODEL_NAME}}::result_t*>(result);
+
+        for (int i = 0; i < N_OUTPUT_SIZE; i++) {
+            result_p[i] = _result[i];
+        }
     }
 };
 

{triggerflow-0.2.5.dist-info → triggerflow-0.2.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: triggerflow
-Version: 0.2.5
+Version: 0.2.7
 Summary: Utilities for ML models targeting hardware triggers
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License

{triggerflow-0.2.5.dist-info → triggerflow-0.2.7.dist-info}/RECORD

@@ -93,10 +93,10 @@ triggerflow/templates/data_types.h,sha256=m7_jLsRixSVjp8auxHZUUNAcSO777C5TTZl__9
 triggerflow/templates/makefile,sha256=A-aetsLC51Bop0T_-yPY8Z8Hg29ApN4YPvKx_jjPuHw,970
 triggerflow/templates/makefile_version,sha256=6kFc_u2oiM9l2rH7RK_BLzdZu1ZEK8PQTQKGBLRY0v4,328
 triggerflow/templates/model-gt.cpp,sha256=qZwuTtsvrKB_mOB-HDb2uOD7mDo4-20EjFiQzRjMdPo,2969
-triggerflow/templates/model_template.cpp,sha256=jMNRcO7NgC6I9Wd2BV3Bim-P1qPsAl_oeVQ8KofQGEw,1807
+triggerflow/templates/model_template.cpp,sha256=Uh80AwDAcFlseonEwV7_HqD85ALs0IPG5JfVT204d5k,1887
 triggerflow/templates/scales.h,sha256=MFcB5S0DEvfzHuUhyZqILR0O4ktugOG-fLnuCDUUewM,373
-triggerflow-0.2.5.dist-info/METADATA,sha256=_psSoh10cZrBvYTjQIKQVw8m4VohIVsjdxlN1MYQazY,5535
-triggerflow-0.2.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-triggerflow-0.2.5.dist-info/entry_points.txt,sha256=5QSV9YDseB_FqgVh9q10BdL4b1I6t68rGwPLXgVL60g,53
-triggerflow-0.2.5.dist-info/top_level.txt,sha256=cX0jkuM9tfxGp002ZBQ1AYgx-6D_NgBtomgPL0WA9bE,43
-triggerflow-0.2.5.dist-info/RECORD,,
+triggerflow-0.2.7.dist-info/METADATA,sha256=vp6187RdnsGgdanBTqvY6jifqZmOZ8JoN92wGT7tBbM,5535
+triggerflow-0.2.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+triggerflow-0.2.7.dist-info/entry_points.txt,sha256=5QSV9YDseB_FqgVh9q10BdL4b1I6t68rGwPLXgVL60g,53
+triggerflow-0.2.7.dist-info/top_level.txt,sha256=cX0jkuM9tfxGp002ZBQ1AYgx-6D_NgBtomgPL0WA9bE,43
+triggerflow-0.2.7.dist-info/RECORD,,