fix: deserializePlugin ignores serialization length

[andrewwhitecdw]

Problem

fix: deserializePlugin ignores serialization length

Fix

Replace:

    nvinfer1::IPluginV2* deserializePlugin(
        char const*,
        void const* serialData,
        size_t serialLength) noexcept override {
        auto* plugin = new FFSGWCVolumePlugin(serialData, serialLength);
        plugin->setPluginNamespace(namespace_.c_str());
        return plugin;
    }

with:

    nvinfer1::IPluginV2* deserializePlugin(
        char const*,
        void const* serialData,
        size_t serialLength) noexcept override {
        if (serialLength != sizeof(GWCParams)) return nullptr;
        auto* plugin = new FFSGWCVolumePlugin(serialData, serialLength);
        plugin->setPluginNamespace(namespace_.c_str());
        return plugin;
    }

Files changed

• cpp/src/gwc_volume_plugin.cpp