提交 1fab1698 编写于 作者: W Wilber 提交者: GitHub

update test=develop (#2416)

add CUDAContext assignment operator to fix cuda compile bug
上级 7a32c431
......@@ -232,6 +232,13 @@ class Context<TargetType::kCUDA> {
std::string name() const { return "CUDAContext"; }
CUDAContext& operator=(const CUDAContext& context) {
this->Init(
context.device_id_, context.exec_stream_id_, context.io_stream_id_);
cublas_fp32_ = const_cast<CUDAContext&>(context).cublas_fp32();
return *this;
}
private:
int device_id_;
// overall information
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册