remove cuda allocator ctor, test=develop (#22212)

revert-22710-feature/integrated_ps_api
Zeng Jinle 5 years ago committed by GitHub
parent 9942d9ed5c
commit 1b76e789cf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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<platform::CUDAPlace>(place)) {}
bool IsAllocThreadSafe() const override;
protected:

Loading…
Cancel
Save