cblas_new
liaogang 8 years ago
parent 48cf64e833
commit 028f3dc4e5

@ -47,6 +47,20 @@ size_t Used<platform::CPUPlace>(platform::CPUPlace place) {
return GetCPUBuddyAllocator()->Used();
}
template <>
void Copy<platform::CPUPlace, platform::CPUPlace>(platform::CPUPlace, void* dst,
platform::CPUPlace, void* src,
size_t size) {
memcpy(dst, src, size);
}
template <>
void Copy<platform::CPUPlace, platform::GPUPlace>(platform::CPUPlace, void* dst,
platform::CPUPlace, void* src,
size_t size) {
memcpy(dst, src, size);
}
#ifndef PADDLE_ONLY_CPU
detail::BuddyAllocator* GetGPUBuddyAllocator(int gpu_id) {

@ -28,5 +28,8 @@ void Free(Place, void*);
template <class Place>
size_t Used(Place);
template <class Place1, class Place2>
void Copy(Place1, void* dst, Place2, void* src, size_t size);
} // namespace memory
} // namespace paddle

Loading…
Cancel
Save