Method unlock

src/backend/common/DefaultMemoryManager.cpp:235–282

}

void DefaultMemoryManager::unlock(void *ptr, bool user_unlock) {
    // Shortcut for empty arrays
    if (!ptr) { return; }

    // Frees the pointer outside the lock.
    uptr_t freed_ptr(nullptr, [this](void *p) { this->nativeFree(p); });
    {
        lock_guard_t lock(this->memory_mutex);
        memory_info &current = this->getCurrentMemoryInfo();

        auto locked_buffer_iter = current.locked_map.find(ptr);
        if (locked_buffer_iter == current.locked_map.end()) {
            // Pointer not found in locked map
            // Probably came from user, just free it
            freed_ptr.reset(ptr);
            return;
        }
        locked_info &locked_buffer_info = locked_buffer_iter->second;
        void *locked_buffer_ptr         = locked_buffer_iter->first;

        if (user_unlock) {
            locked_buffer_info.user_lock = false;
        } else {
            locked_buffer_info.manager_lock = false;
        }

        // Return early if either one is locked
        if (locked_buffer_info.user_lock || locked_buffer_info.manager_lock) {
            return;
        }

        size_t bytes = locked_buffer_info.bytes;
        current.lock_bytes -= locked_buffer_info.bytes;
        current.lock_buffers--;

        if (this->debug_mode) {
            // Just free memory in debug mode
            if (locked_buffer_info.bytes > 0) {
                freed_ptr.reset(locked_buffer_ptr);
                current.total_buffers--;
                current.total_bytes -= locked_buffer_info.bytes;
            }
        } else {
            current.free_map[bytes].emplace_back(ptr);
        }
        current.locked_map.erase(locked_buffer_iter);
    }
}

void DefaultMemoryManager::signalMemoryCleanup() {
    cleanDeviceMemoryManager(this->getActiveDeviceId());