#include <ExecutionObservers.h>

Collaboration diagram for onert::exec::TracingObserver:

Public Member Functions
	TracingObserver (const std::string &workspace_dir, const ir::Graph &graph, const util::TracingCtx *tracing_ctx)

	~TracingObserver ()

void	handleSubgraphBegin (std::pair< ir::ModelIndex, ir::SubgraphIndex >) override
	Invoked just before model (not individual operation) execution begins.

void	handleJobBegin (IExecutor , std::pair< ir::ModelIndex, ir::SubgraphIndex >, ir::OperationIndex, const backend::Backend ) override

void	handleJobEnd (IExecutor , std::pair< ir::ModelIndex, ir::SubgraphIndex >, ir::OperationIndex, const backend::Backend ) override

void	handleSubgraphEnd (std::pair< ir::ModelIndex, ir::SubgraphIndex >) override
	Invoked just after model (not individual operation) execution ends.

ObserverType	type () const override

Public Member Functions inherited from onert::exec::IExecutionObserver
virtual	~IExecutionObserver ()=default

Detailed Description

Definition at line 104 of file ExecutionObservers.h.

Constructor & Destructor Documentation

◆ TracingObserver()

onert::exec::TracingObserver::TracingObserver	(	const std::string &	workspace_dir,
		const ir::Graph &	graph,
		const util::TracingCtx *	tracing_ctx
	)

Definition at line 117 of file ExecutionObservers.cc.

  : _recorder{std::make_unique<EventRecorder>()}, _collector{_recorder.get()}, _graph{graph},
    _workspace_dir{workspace_dir}, _tracing_ctx{tracing_ctx}, _triggered{false}
{
  // DO NOTHING
}

◆ ~TracingObserver()

onert::exec::TracingObserver::~TracingObserver ( )

Definition at line 125 of file ExecutionObservers.cc.

{
  try
  {
    // Write file if this observer is triggered at least once
    if (_triggered)
    {
      auto event_writer = EventWriter::get(_workspace_dir);
      event_writer->startToUse();
      event_writer->readyToFlush(std::move(_recorder));
    }
  }
  catch (const std::exception &e)
  {
    std::cerr << "E: Fail to record event in TracingObserver: " << e.what() << std::endl;
  }
}

References EventWriter::get().

Member Function Documentation

◆ handleJobBegin()

void onert::exec::TracingObserver::handleJobBegin	(	IExecutor *	,
		std::pair< ir::ModelIndex, ir::SubgraphIndex >	ind,
		ir::OperationIndex	op_ind,
		const backend::Backend *	backend
	)

overridevirtual

Implements onert::exec::IExecutionObserver.

Definition at line 151 of file ExecutionObservers.cc.

{
  std::string backend_id = backend->config()->id();
  const auto &op = _graph.operations().at(op_ind);
  auto ev = EventCollector::OpSeqEvent{_tracing_ctx,      EventCollector::Edge::BEGIN,
                                       ind.first.value(), ind.second.value(),
                                       backend_id,        op_ind.value(),
                                       op.name()};
  // add shape of inputs
  setUserData(_graph, &op, ev.userData);
  _collector.onEvent(ev);
}

References onert::util::ObjectManager< Index, Object >::at(), EventCollector::BEGIN, onert::backend::Backend::config(), onert::ir::IOperation::name(), EventCollector::onEvent(), onert::ir::Graph::operations(), and onert::util::Index< T, DummyTag >::value().

◆ handleJobEnd()

void onert::exec::TracingObserver::handleJobEnd	(	IExecutor *	,
		std::pair< ir::ModelIndex, ir::SubgraphIndex >	ind,
		ir::OperationIndex	op_ind,
		const backend::Backend *	backend
	)

overridevirtual

Implements onert::exec::IExecutionObserver.

Definition at line 165 of file ExecutionObservers.cc.

{
  std::string backend_id = backend->config()->id();
  _collector.onEvent(EventCollector::OpSeqEvent{
    _tracing_ctx, EventCollector::Edge::END, ind.first.value(), ind.second.value(), backend_id,
    op_ind.value(), _graph.operations().at(op_ind).name()});
}

References onert::util::ObjectManager< Index, Object >::at(), onert::backend::Backend::config(), EventCollector::END, EventCollector::onEvent(), onert::ir::Graph::operations(), and onert::util::Index< T, DummyTag >::value().

◆ handleSubgraphBegin()

void onert::exec::TracingObserver::handleSubgraphBegin ( std::pair< ir::ModelIndex, ir::SubgraphIndex > )

overridevirtual

Invoked just before model (not individual operation) execution begins.

Reimplemented from onert::exec::IExecutionObserver.

Definition at line 143 of file ExecutionObservers.cc.

{
  _triggered = true;
 
  _collector.onEvent(EventCollector::SubgEvent{_tracing_ctx, EventCollector::Edge::BEGIN,
                                               ind.first.value(), ind.second.value()});
}

References EventCollector::BEGIN, and EventCollector::onEvent().

◆ handleSubgraphEnd()

void onert::exec::TracingObserver::handleSubgraphEnd ( std::pair< ir::ModelIndex, ir::SubgraphIndex > )

overridevirtual

Invoked just after model (not individual operation) execution ends.

Reimplemented from onert::exec::IExecutionObserver.

Definition at line 174 of file ExecutionObservers.cc.

{
  _collector.onEvent(EventCollector::SubgEvent{_tracing_ctx, EventCollector::Edge::END,
                                               ind.first.value(), ind.second.value()});
}

References EventCollector::END, and EventCollector::onEvent().