提交 454db666 编写于 作者: M minqiyang

Accelerate lstm

上级 a3fa3f85
...@@ -180,7 +180,7 @@ class ConcatFunctor<platform::CUDADeviceContext, T> { ...@@ -180,7 +180,7 @@ class ConcatFunctor<platform::CUDADeviceContext, T> {
} }
// Wait() must be called because `inputs_data` may be destructed before // Wait() must be called because `inputs_data` may be destructed before
// kernel ends // kernel ends
context.Wait(); /* context.Wait(); */
} }
}; };
...@@ -258,7 +258,7 @@ class SplitFunctor<platform::CUDADeviceContext, T> { ...@@ -258,7 +258,7 @@ class SplitFunctor<platform::CUDADeviceContext, T> {
} }
// Wait() must be called because `outputs_data` may be destructed before // Wait() must be called because `outputs_data` may be destructed before
// kernel ends // kernel ends
context.Wait(); /* context.Wait(); */
} }
}; };
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册