index a289507d4b050a9d9418b8424de26397e16b7f87..2291d97f9ab543aed19d147b7464b2742cd937fc 100644 (file)
Executor* CreateExecutor(DeviceType dt, int num, const Configuration& c,
int layer_group_id);
-void AllocateMemory(const vector<EOP *>& EOPs);
-void FreeMemory (const vector<EOP *>& EOPs);
-
int main(int argc, char *argv[])
{
return new Executor(dt, ids, c, layer_group_id);
}
-// Allocate input and output memory for each EO
-void AllocateMemory(const vector<EOP *>& EOPs)
-{
- // Allocate input and output buffers for each execution object
- for (auto eop : EOPs)
- {
- size_t in_size = eop->GetInputBufferSizeInBytes();
- size_t out_size = eop->GetOutputBufferSizeInBytes();
- ArgInfo in = { ArgInfo(malloc(in_size), in_size)};
- ArgInfo out = { ArgInfo(malloc(out_size), out_size)};
- eop->SetInputOutputBuffer(in, out);
- }
-}
-
-// Free the input and output memory associated with each EO
-void FreeMemory(const vector<EOP *>& EOPs)
-{
- for (auto eop : EOPs)
- {
- free(eop->GetInputBufferPtr());
- free(eop->GetOutputBufferPtr());
- }
-
-}