Method alloc_direct

src/core/impl/graph/var_node_mem_mgr/defrag.cpp:8–18

using namespace cg;

void VarDevMemDefragmenter::alloc_direct(
        VarNode* var, DeviceTensorStorage& storage, size_t size) {
    if (!storage.comp_node_valid())
        storage.comp_node(var->comp_node());
    if (size > storage.size()) {
        m_mem_mgr->static_device_memory_manager()->allocator().alloc_dynamic(
                var, storage, size);
        storage.ptr();  // apply lazy alloc
    }
    mgb_assert(storage.size() >= size);
}

#if MGB_ENABLE_VAR_DEV_MEM_DEFRAGMENTER