pytorch/caffe2/predictor/emulator/emulator.h
James Sun 68843c683d Open source multithreaded predictor bench utils (#11135)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/11135

This diff does not have any logic change; it simply move files/functions/classes around.
Open source (almost all) necessary dependency for multithreaded predictor bench.
The benchmark itself can be open sourced once the predictor is open sourced.

Reviewed By: salexspb

Differential Revision: D9602006

fbshipit-source-id: 386c9483e2c64c8b7d36e4600189c4e0b7e159ff
2018-10-18 14:16:36 -07:00

21 lines
354 B
C++

#pragma once
#include "caffe2/core/logging.h"
namespace caffe2 {
namespace emulator {
/*
* A net emulator. In short, it can run nets with given @iterations.
*/
class Emulator {
public:
virtual void init() = 0;
virtual void run(const uint64_t iterations) = 0;
virtual ~Emulator() noexcept {}
};
} // namespace emulator
} // namespace caffe2