Method memoryUsed

test/expr/ModuleTest.cpp:1231–1265

class WinogradMemoryTest : public MNNTestCase {
public:
    float memoryUsed(int level) {
        auto x = _Input({1, 64, 224, 224}, MNN::Express::NC4HW4, halide_type_of<float>());
        x->setName("Input");
        auto y = _Conv(0.0f, 0.0f, x, {64, 112}, {3, 3});
        y->setName("Prob");
        std::unique_ptr<MNN::NetT> net(new NetT);
        Variable::save({y}, net.get());
        y = nullptr;
        flatbuffers::FlatBufferBuilder builderOutput(1024);
        auto len = MNN::Net::Pack(builderOutput, net.get());
        builderOutput.Finish(len);
        int sizeOutput    = builderOutput.GetSize();
        auto bufferOutput = builderOutput.GetBufferPointer();
        // Force use CPU Runtime
        BackendConfig bnConfig;
        auto exe = Executor::newExecutor(MNN_FORWARD_CPU, bnConfig, 1);
        ExecutorScope scope(exe);
        Module::Config config;
        config.shapeMutable = false;
        std::shared_ptr<Module> interp0;
        
        MNN::ScheduleConfig sconfig;
        sconfig.numThread = 1;
        std::vector<MNN::ScheduleConfig> sconfigs = {sconfig};
        auto rtInfo = Express::ExecutorScope::Current()->getRuntime();
        auto rt = rtInfo.first.begin()->second;
        std::shared_ptr<Executor::RuntimeManager> rtMgr(Executor::RuntimeManager::createRuntimeManager(sconfigs));
        rtMgr->setMode(Interpreter::Session_Memory_Collect);
        rtMgr->setHint(Interpreter::WINOGRAD_MEMORY_LEVEL, level);
        config.rearrange = false; // When set WINOGRAD_MEMORY_LEVEL=0 to test memory, must set rearrange=false.
        interp0.reset(Module::load({"Input"}, {"Prob"}, bufferOutput, sizeOutput, rtMgr, &config), Module::destroy);
        float memoryInMB = 0.0f;
        rtMgr->getInfo(Interpreter::MEMORY, &memoryInMB);
        return memoryInMB;
    }
    virtual bool run(int precision) {
        float mem0 = memoryUsed(0);
        float mem1 = memoryUsed(1);