[ssd_multibox] Addressing review comments
[tidl/tidl-api.git] / examples / one_eo_per_frame / main.cpp
index a7f10a8b106bddc79b58ed8efc199d4c78bad980..8872c06438d3aed18c54d8ebbcbd0bcf6fa320cd 100644 (file)
@@ -53,9 +53,6 @@ bool Run(const string& config_file, int num_eve,int num_dsp,
 Executor* CreateExecutor(DeviceType dt, int num, const Configuration& c);
 void      CollectEOs(const Executor *e, vector<ExecutionObject *>& EOs);
 
-void AllocateMemory(const vector<ExecutionObject *>& EOs);
-void FreeMemory    (const vector<ExecutionObject *>& EOs);
-
 
 int main(int argc, char *argv[])
 {
@@ -77,6 +74,11 @@ int main(int argc, char *argv[])
 
     unique_ptr<const char> reference_output(ReadReferenceOutput(ref_file));
 
+    // Enable time stamp generation. The timestamp file is post processed
+    // by execution_graph.py to generate graphical view of frame execution.
+    // Refer to the User's Guide for details.
+    EnableTimeStamps("1eo.log");
+
     bool status = Run(config_file, num_eve, num_dsp, reference_output.get());
 
     if (!status)
@@ -96,7 +98,7 @@ bool Run(const string& config_file, int num_eve, int num_dsp,
     if (!c.ReadFromFile(config_file))
         return false;
 
-    // heap sizes determined using Configuration::showHeapStats
+    // Heap sizes for this network determined using Configuration::showHeapStats
     c.PARAM_HEAP_SIZE   = (3 << 20); // 3MB
     c.NETWORK_HEAP_SIZE = (20 << 20); // 20MB
 
@@ -174,27 +176,3 @@ void CollectEOs(const Executor *e, vector<ExecutionObject *>& EOs)
         EOs.push_back((*e)[i]);
 }
 
-// Allocate input and output memory for each EO
-void AllocateMemory(const vector<ExecutionObject *>& EOs)
-{
-    // Allocate input and output buffers for each execution object
-    for (auto eo : EOs)
-    {
-        size_t in_size  = eo->GetInputBufferSizeInBytes();
-        size_t out_size = eo->GetOutputBufferSizeInBytes();
-        ArgInfo in  = { ArgInfo(malloc(in_size),  in_size)};
-        ArgInfo out = { ArgInfo(malloc(out_size), out_size)};
-        eo->SetInputOutputBuffer(in, out);
-    }
-}
-
-// Free the input and output memory associated with each EO
-void FreeMemory(const vector<ExecutionObject *>& EOs)
-{
-    for (auto eo : EOs)
-    {
-        free(eo->GetInputBufferPtr());
-        free(eo->GetOutputBufferPtr());
-    }
-
-}