1 #ifndef CAFFE2_CORE_NET_ASYNC_DAG_GPU_H_ 2 #define CAFFE2_CORE_NET_ASYNC_DAG_GPU_H_ 4 #include "caffe2/core/common.h" 5 #include "caffe2/core/net_dag.h" 6 #include "caffe2/core/workspace.h" 7 #include "caffe2/proto/caffe2.pb.h" 20 bool SupportsAsync()
override {
23 bool RunAt(
int chain_id,
const std::vector<int>& chain)
override;
26 bool DoRunAsync()
override;
30 std::vector<int32_t> eventRecorded_;
32 int stream(
const DeviceOption& device_option);
33 static thread_local std::vector<int> stream_counters_;
40 #endif // CAFFE2_CORE_NET_ASYNC_DAG_GPU_H_ Workspace is a class that holds all the related objects created during runtime: (1) all blobs...
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...