Method run

test/expr/ModuleTest.cpp:1618–1687

class InputModuleTest : public MNNTestCase {
public:
    virtual bool run(int precision) {
        auto executor = cloneCurrentExecutor();
        ExecutorScope scope(executor);
        auto y = _mobileNetV1Expr(nullptr, false);
        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();
        auto test = [&](bool shapeMutable) {
            Module::Config config;
            config.shapeMutable = shapeMutable;
            config.rearrange = true;
            std::shared_ptr<Module> m0;
            std::shared_ptr<Module> m1;
            std::shared_ptr<Module> m2;
            {
                MNN::ScheduleConfig sconfig;
                sconfig.numThread = 1;
                MNN::BackendConfig bnconfig;
                bnconfig.precision = MNN::BackendConfig::Precision_Low;
                sconfig.backendConfig = &bnconfig;
                std::vector<MNN::ScheduleConfig> sconfigs = {sconfig};
                std::shared_ptr<Executor::RuntimeManager> rtMgr(Executor::RuntimeManager::createRuntimeManager(sconfigs));
                m0.reset(Module::load({"Input"}, {"Prob"}, bufferOutput, sizeOutput, rtMgr, &config), Module::destroy);
                bnconfig.precision = MNN::BackendConfig::Precision_Normal;
                std::shared_ptr<Executor::RuntimeManager> rtMgr2(Executor::RuntimeManager::createRuntimeManager(sconfigs));
                m1.reset(Module::load({"Input"}, {"Prob"}, bufferOutput, sizeOutput, rtMgr2, &config), Module::destroy);
                m2.reset(Module::load({"Input"}, {"Prob"}, bufferOutput, sizeOutput), Module::destroy);
            }
            auto x = _Input({1, 3, 32, 32}, NCHW, halide_type_of<float>());
            auto ptr = x->writeMap<float>();
            for (int i=0; i<x->getInfo()->size; ++i) {
                ptr[i] = 1.0f * i;
            }
            x = x + x;
            auto prob = m0->onForward({x})[0];
            auto pptr = prob->readMap<float>();
            
            float s0 = _ReduceSum(m0->onForward({x})[0])->readMap<float>()[0];
            float s1 = _ReduceSum(m1->onForward({x})[0])->readMap<float>()[0];
            float s2 = _ReduceSum(m2->onForward({x})[0])->readMap<float>()[0];
            // Normally s2 is correct, compare to s2
            if (fabsf(s0-s2) / s2 > 0.2f) {
                FUNC_PRINT_ALL(s0, f);
                FUNC_PRINT_ALL(s2, f);
                return false;
            }
            if (fabsf(s1-s2) / s2 > 0.2f) {
                FUNC_PRINT_ALL(s1, f);
                FUNC_PRINT_ALL(s2, f);
                return false;
            }
            return true;
        };