Method run

test/expr/ModuleTest.cpp:1459–1503

class ExecutorResetLoadModuleTest : public MNNTestCase {
public:
    virtual bool run(int precision) {
        BackendConfig originConfig;
        auto exe = Executor::newExecutor(MNN_FORWARD_CPU, originConfig, 1);
        ExecutorScope _s(exe);
        // Make Model include convolution in shape compute and content compute
        auto x = _Input({1, 3, 24, 24}, NCHW, halide_type_of<float>());
        x->setName("x");
        auto xs = _Convert(_Reshape(_Cast<float>(_Shape(x, NCHW)), {1, 1, 2, 2}), NC4HW4);
        xs = _Convert(_Conv(1.0f, 0.0f, xs, {1, 1}, {2, 2}), NCHW);
        auto y = _Conv(0.1f, 0.0f, _Convert(x, NC4HW4), {3, 1}, {3, 3});
        y = _Convert(y, NCHW);
        y = _ReduceMean(y);
        y = y * _Reciprocal(xs);
        auto info = y->getInfo();
        y->setName("y");
        auto buffer = Variable::save({y});
        MNN::ScheduleConfig sconfig;
        BackendConfig bnConfig;
        bnConfig.precision = MNN::BackendConfig::Precision_Low;
        bnConfig.memory = MNN::BackendConfig::Memory_Low;
        sconfig.backendConfig = &bnConfig;
        sconfig.numThread = 4;
        exe->setGlobalExecutorConfig(MNN_FORWARD_CPU, bnConfig, 4);
        std::shared_ptr<Executor::RuntimeManager> rtMgr(Executor::RuntimeManager::createRuntimeManager(sconfig));
        Module::Config config;
        config.rearrange = false;
        std::shared_ptr<MNN::Express::Module> m0(Module::load({"x"}, {"y"}, (const unsigned char*)buffer.data(), buffer.size(), nullptr, &config), Module::destroy);
        config.rearrange = true;
        std::shared_ptr<MNN::Express::Module> m1(Module::load({"x"}, {"y"}, (const unsigned char*)buffer.data(), buffer.size(), rtMgr, &config), Module::destroy);
        auto m0Rt = m0->getInfo()->runTimeManager;
        auto m1Rt = m1->getInfo()->runTimeManager;
        if (nullptr == m0Rt->getBnConfig() || nullptr == m1Rt->getBnConfig()) {
            FUNC_PRINT(1);
            return false;
        }
        if (MNN::BackendConfig::Precision_Low != m0Rt->getBnConfig()->precision || MNN::BackendConfig::Memory_Low != m0Rt->getBnConfig()->memory) {
            FUNC_PRINT(1);
            return false;
        }
        if (MNN::BackendConfig::Precision_Low != m1Rt->getBnConfig()->precision || MNN::BackendConfig::Memory_Low != m1Rt->getBnConfig()->memory) {
            FUNC_PRINT(1);
            return false;
        }
        return true;
    }
};
MNNTestSuiteRegister(ExecutorResetLoadModuleTest, "expr/ExecutorResetLoadModuleTest");