Merge pull request #4 from cms-hls4ml/utility

add utility
cms-hls4ml · Mar 16, 2023 · 74771a9 · 74771a9
2 parents a658b8a + 041bb52
commit 74771a9
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/AXOL1TL_v1/GTADModel_emulator_v1.cpp b/AXOL1TL_v1/GTADModel_emulator_v1.cpp
@@ -3,6 +3,7 @@
 #include "NN/nnet_utils/nnet_common.h"
 #include <any>
 #include <array>
+#include <utility>
 #include "ap_fixed.h"
 #include "ap_int.h"
 #include "scales.h"
@@ -63,11 +64,11 @@ class GTADModel_emulator_v1 : public hls4mlEmulator::Model {
     // return results as an std::pair
     // first = reconstructed output
     // second = loss
-    std::pair<std::array<result_t, N_LAYER_6>, resultsq_t> result_p = std::any_cast<std::pair<std::array<result_t, N_LAYER_6>, resultsq_t>>(result);
+    std::pair<std::array<result_t, N_LAYER_6>, resultsq_t> *result_p = std::any_cast<std::pair<std::array<result_t, N_LAYER_6>, resultsq_t>*>(result);
     for (int i = 0; i < N_LAYER_6; i++) {
-      result_p.first[i] = _result[i];
+      result_p->first[i] = _result[i];
     }
-    result_p.second = _loss;
+    result_p->second = _loss;
   }
 
 };