emailweixu-patch-1
dzhwinter 7 years ago
parent f18f3826dc
commit 17b1c369b1

@ -116,6 +116,8 @@ inline T *Vector<T>::mutable_data(platform::Place place) {
this->size() * sizeof(T), ctx->stream());
ctx->Wait();
return static_cast<T *>(cuda_ptr_.get());
#else
return nullptr;
#endif
} else {
PADDLE_THROW("Unsupport Place.");

Loading…
Cancel
Save