diff --git a/examples/test/main.cpp b/examples/test/main.cpp
index 4d6f2b7ea00ca48f62055c5c1c611973eab0357e..2233dad413228c6954ee3323e686a529f354e43a 100644 (file)
--- a/examples/test/main.cpp
+++ b/examples/test/main.cpp
assert (input_data_file.good());
assert (output_data_file.good());
- // Determine input frame size from configuration
- size_t frame_sz = configuration.inWidth * configuration.inHeight *
- configuration.inNumChannels;
-
try
{
// Create a executor with the approriate core type, number of cores
std::vector<void *> buffers;
for (auto &eo : execution_objects)
{
- ArgInfo in = { ArgInfo(malloc(frame_sz), frame_sz)};
- ArgInfo out = { ArgInfo(malloc(frame_sz), frame_sz)};
+ 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);
buffers.push_back(in.ptr());