提交 d34e385d 编写于 作者: Z zjd1988 提交者: Xinran Xu

assert locator.device < sd.MAX_NR_DEVICE when bool(avaiable_node) = true

Signed-off-by: Nzjd1988 <942002795@qq.com>
上级 502d6b89
......@@ -659,10 +659,10 @@ CompNode::Impl* CudaCompNode::load_cuda(
if (!available_node) {
mgb_assert(sd.nr_node < sd.MAX_NR_COMP_NODE,
"too many CompNode allocated");
mgb_assert(locator.device < sd.MAX_NR_COMP_NODE,
"device number too large");
available_node = &sd.node[sd.nr_node ++];
}
mgb_assert(locator.device < sd.MAX_NR_DEVICE,
"device number too large");
mgb_assert(!available_node->m_initialized);
available_node->init(locator, locator_logical);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册