Method createKVCacheFile

source/core/KVCacheManager.cpp:27–43

}

void KVCacheManager::createKVCacheFile(std::string keyPath, std::string valuePath) {
    // Each layer has its own kvcache, so we have to create a key file and a value file for each layer and the file name must be unique
    // Here we use the address of the mResource as the file name because the addresses of mResource in different layers are guaranteed to be different
    std::string fileName = addrToHex(this);
    mBaseFileName = MNNFilePathConcat(mConfig.mKVCacheDir, fileName);

    std::string pathk    = keyPath.size() > 0 ? keyPath : mBaseFileName + ".k";
    std::string pathv    = valuePath.size() > 0 ? valuePath : mBaseFileName + ".v";
    mKeyCacheFD   = MNNCreateFile(pathk.c_str());
    mValueCacheFD = MNNCreateFile(pathv.c_str());
    if (mKeyCacheFD == INVALID_FILE) {
        MNN_PRINT("Failed to create the file: %s\n", pathk.c_str());
    }
    if (mValueCacheFD == INVALID_FILE) {
        MNN_PRINT("Failed to create the file: %s\n", pathv.c_str());
    }
}

void KVCacheManager::removeKVCacheFile() {
    std::string pathk    = mBaseFileName + ".k";