Method run

test/expr/ModuleTest.cpp:1409–1453

class PrearrangeTest : public MNNTestCase {
public:
    virtual bool run(int precision) {
        // 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;
        sconfig.backendConfig = &bnConfig;
        auto exe = Executor::newExecutor(MNN_FORWARD_CPU, bnConfig, 4);
        ExecutorScope scope(exe);
        std::vector<MNN::ScheduleConfig> sconfigs = {sconfig};
        std::shared_ptr<Executor::RuntimeManager> rtMgr(Executor::RuntimeManager::createRuntimeManager(sconfigs));
        rtMgr->setMode(Interpreter::Session_Memory_Collect);
        Module::Config config;
        config.rearrange = false;
        std::shared_ptr<MNN::Express::Module> m0(Module::load({"x"}, {"y"}, (const unsigned char*)buffer.data(), buffer.size(), rtMgr, &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 size = x->getInfo()->size;
        auto xPtr = x->writeMap<float>();
        for (int v=0; v<size; ++v) {
            xPtr[v] = 0.01f;
        }
        auto y0 = m0->onForward({x})[0]->readMap<float>()[0];
        auto y1 = m1->onForward({x})[0]->readMap<float>()[0];
        if (fabsf(y0 - y1) > 0.000001f) {
            return false;
        }
        rtMgr->setExternalPath(".", Interpreter::EXTERNAL_FEATUREMAP_DIR);
        std::shared_ptr<MNN::Express::Module> m2(Module::load({"x"}, {"y"}, (const unsigned char*)buffer.data(), buffer.size(), rtMgr, &config), Module::destroy);
        auto y2 = m2->onForward({x})[0]->readMap<float>()[0];
        if (fabsf(y0 - y2) > 0.000001f) {
            return false;
        }
        return true;
    }
};
MNNTestSuiteRegister(PrearrangeTest, "expr/PrearrangeTest");