Browse Source

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

Signed-off-by: zjd1988 <942002795@qq.com>
tags/v0.5.0
zjd1988 Xinran Xu 5 years ago
parent
commit
d34e385dd7
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      src/core/impl/comp_node/cuda/comp_node.cpp

+ 2
- 2
src/core/impl/comp_node/cuda/comp_node.cpp View File

@@ -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);


Loading…
Cancel
Save