|
|
@ -14,12 +14,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
#pragma once
|
|
|
|
#pragma once
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#include <memory>
|
|
|
|
#include <string>
|
|
|
|
#include <string>
|
|
|
|
#include <vector>
|
|
|
|
#include <vector>
|
|
|
|
|
|
|
|
|
|
|
|
#include "ThreadPool.h"
|
|
|
|
#include "ThreadPool.h"
|
|
|
|
|
|
|
|
#include "paddle/fluid/framework/details/fast_threaded_ssa_graph_executor.h"
|
|
|
|
#include "paddle/fluid/framework/details/multi_devices_helper.h"
|
|
|
|
#include "paddle/fluid/framework/details/multi_devices_helper.h"
|
|
|
|
#include "paddle/fluid/framework/details/threaded_ssa_graph_executor.h"
|
|
|
|
|
|
|
|
#include "paddle/fluid/framework/ir/graph.h"
|
|
|
|
#include "paddle/fluid/framework/ir/graph.h"
|
|
|
|
|
|
|
|
|
|
|
|
namespace paddle {
|
|
|
|
namespace paddle {
|
|
|
@ -48,7 +48,8 @@ class ParallelSSAGraphExecutor : public SSAGraphExecutor {
|
|
|
|
std::vector<platform::Place> places_;
|
|
|
|
std::vector<platform::Place> places_;
|
|
|
|
std::vector<std::unique_ptr<ir::Graph>> graphs_;
|
|
|
|
std::vector<std::unique_ptr<ir::Graph>> graphs_;
|
|
|
|
|
|
|
|
|
|
|
|
std::vector<std::unique_ptr<details::ThreadedSSAGraphExecutor>> executors_;
|
|
|
|
std::vector<std::unique_ptr<details::FastThreadedSSAGraphExecutor>>
|
|
|
|
|
|
|
|
executors_;
|
|
|
|
ExceptionHolder exception_holder_;
|
|
|
|
ExceptionHolder exception_holder_;
|
|
|
|
};
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|