diff --git a/paddle/fluid/memory/allocation/cuda_allocator.h b/paddle/fluid/memory/allocation/cuda_allocator.h index ae48d26ceeb5a4611d4e140e3851eee5f11b4d5b..5969d4d20ddee498c8257412cc9d8fa7177d475b 100644 --- a/paddle/fluid/memory/allocation/cuda_allocator.h +++ b/paddle/fluid/memory/allocation/cuda_allocator.h @@ -24,8 +24,7 @@ namespace allocation { class CUDAAllocator : public Allocator { public: explicit CUDAAllocator(const platform::CUDAPlace& place) : place_(place) {} - explicit CUDAAllocator(const platform::Place& place) - : place_(boost::get(place)) {} + bool IsAllocThreadSafe() const override; protected: