feat[shared_library]: shared library works in python but not with jupyter

Author: Djokzer

hls4ml/templates/vitis_accelerator/libs/DataBatcher.hpp

@@ -251,6 +251,27 @@ template <class T, class U> class DataBatcher {
         profilingResultsDump.clear();
     }
 
+    /**
+     * \brief Returns results to a shared buffer.
+     * \param data_out Pointer to the shared buffer where results will be written
+     * \param size Size of the shared buffer in bytes
+     */
+    void returnSharedResults(double* data_out, uint64_t size) {
+        if (size == 0 || data_out == nullptr) {
+            throw std::runtime_error("No data to return");
+        }
+
+        if (size != storedEvalResults.size()) {
+            throw std::runtime_error("Size mismatch: " + std::to_string(size) + " != " + std::to_string(storedEvalResults.size()));
+        }
+
+        for (uint64_t i = 0; i < size; i++) {
+            data_out[i] = static_cast<double>(storedEvalResults[i]);
+        }
+
+        std::cout << "Returning " << size << " results to memory buffer" << std::endl;
+    }
+
     uint64_t getSampleCount() { return originalSampleCount; }
 
     uint64_t getPaddedSampleCount() { return numBatches * _batchsize; }

hls4ml/templates/vitis_accelerator/libs/FpgaObj.hpp

@@ -232,6 +232,18 @@ template <class T, class U> class FpgaObj {
         db->closeFile();
     }
 
+    /**
+     * \brief Returns results to a shared buffer.
+     * \param data_out Pointer to the shared buffer where results will be written
+     * \param size Size of the shared buffer in bytes
+     */
+    void returnSharedResults(double* data_out, uint64_t size) {
+        if (db == nullptr) {
+            throw std::runtime_error("No data loaded");
+        }
+        db->returnSharedResults(data_out, size);
+    }
+
   private:
     int _batchsize;
     int _sampleInputSize;

hls4ml/templates/vitis_accelerator/myproject_host_cl.cpp

@@ -23,7 +23,7 @@ extern "C" void predict(double *input, uint64_t input_size, double *output, uint
 
     fpga.checkResults(params.referenceFilename);
 
-    fpga.saveResults(params.outputFilename);
+    fpga.returnSharedResults(output, output_size);
 }
 
 int main(int argc, char **argv) {