提交 2fc7fc7a 编写于 作者: Y Yang Yang

pass multiple forward backward

上级 e3161bb6
...@@ -279,8 +279,10 @@ TEST_F(ExecutorTesterRandom, GPU) { ...@@ -279,8 +279,10 @@ TEST_F(ExecutorTesterRandom, GPU) {
std::unique_ptr<Executor> executor(new Executor(places)); std::unique_ptr<Executor> executor(new Executor(places));
executor->Run(init_pdesc_, GetGlobalScope(), 0); executor->Run(init_pdesc_, GetGlobalScope(), 0);
executor->Run(pdesc_, GetGlobalScope(), 0); for (int batch_id = 0; batch_id < 3; batch_id++) {
std::vector<std::vector<float>> result = GetFetchVariable<float>(); executor->Run(pdesc_, GetGlobalScope(), 0);
std::vector<std::vector<float>> result = GetFetchVariable<float>();
}
} }
TEST_F(ExecutorTesterFeedAndFetch, GPU) { TEST_F(ExecutorTesterFeedAndFetch, GPU) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册