inferNode.cpp 840 B

1234567891011121314151617181920212223242526272829303132
  1. #include "nodes/base/base.hpp"
  2. #include "nodes/infer/inferNode.hpp"
  3. #include <unordered_map>
  4. namespace Node
  5. {
  6. void InferNode::work()
  7. {
  8. printf("InferNode %s\n", name_.c_str());
  9. while (running_)
  10. {
  11. for (auto& input_buffer : input_buffers_)
  12. {
  13. std::shared_ptr<meta::MetaData> metaData;
  14. if (!input_buffer.second->try_pop(metaData))
  15. {
  16. continue;
  17. }
  18. printf("Node %s get data from %s\n", name_.c_str(), input_buffer.first.c_str());
  19. // do something
  20. for (auto& output_buffer : output_buffers_)
  21. {
  22. printf("Node %s push data to %s\n", name_.c_str(), output_buffer.first.c_str());
  23. output_buffer.second->push(metaData);
  24. }
  25. }
  26. }
  27. };
  28. } // namespace Node