未验证 提交 1b76e789 编写于 作者: Z Zeng Jinle 提交者: GitHub

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

上级 9942d9ed
......@@ -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:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册