ipc_evaluator.h

Go to the documentation of this file.

#pragma once
#include <iostream>
#include <map>
#include <memory>
#include <stdexcept>
 
#include "computation_graph/public_base_op.h"
#include "computation_graph/raw_memory_object.h"
#include "ml_core/callbacks.h" // for ProgressCallback
#include <filesystem>
#include <functional>
#include <list>
#include <thread>
 
namespace blace {
namespace ipc {
 
class IpcEvaluatorImpl;
 
class EXPORT_OR_IMPORT IpcEvaluator {
public:
  IpcEvaluator(::blace::ops::OpP computation_graph);
 
  std::pair<ml_core::ReturnCode, std::shared_ptr<RawMemoryObject>>
  evaluateToRawMemory(
      std::shared_ptr<ml_core::ProgressCallback> progress_callback = nullptr,
      int timeout_in_seconds = 30);
 
private:
  ::blace::ops::OpP computation_graph;
  std::shared_ptr<IpcEvaluatorImpl> impl;
};
 
} // namespace ipc
} // namespace blace