提交 57b973ae 编写于 作者: Y Yang Yu

Merge branch 'feature/is_nan' into feature/check_nan_executor

......@@ -64,7 +64,7 @@ struct CastToPyBufferImpl<true, I, ARGS...> {
tensor.dims(), platform::CPUPlace()));
platform::DeviceContextPool &pool =
platform::DeviceContextPool::Instance;
platform::DeviceContextPool::Instance();
auto dev_ctx = static_cast<const platform::CUDADeviceContext *>(
pool.Get(tensor.place()));
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册