Merge branch 'feature/enhance_dev_ctx_pool' into feature/is_nan

del_some_in_makelist
Yang Yu 7 years ago
commit 71157b33ca

@ -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()));

Loading…
Cancel
Save