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