Method run

test/expr/RasterOutputTest.cpp:37–75

class RasterOutputTest : public MNNTestCase {
public:
    virtual bool run(int precision) {
        auto executor = cloneCurrentExecutor();
        ExecutorScope scope(executor);
        auto net = _createModel();
        auto x = _Input({1, 3, 224, 224}, NCHW, halide_type_of<int>());
        auto y = _Transpose(x, {0, 1, 3, 2});
        auto z = _Add(y, _Scalar<int>(1));
        auto q = _Negative(y);
        auto p = _Transpose(q, {0, 3, 1, 2});
        {
            auto xPtr = x->writeMap<int>();
            for (int v = 0; v < 1 * 3 * 224 * 224; ++ v) {
                xPtr[v] = v;
            }
            x->unMap();
        }
        auto outputs = net->onForward({x});
        {
            auto dPtr = outputs[0]->readMap<int>();
            auto zPtr = z->readMap<int>();
            auto size = z->getInfo()->size;
            for (int v = 0; v < size; ++v) {
                if (zPtr[v] != dPtr[v]) {
                    return false;
                }
            }
        }
        {
            auto dPtr = outputs[1]->readMap<int>();
            auto zPtr = p->readMap<int>();
            auto size = p->getInfo()->size;
            for (int v = 0; v < size; ++v) {
                if (zPtr[v] != dPtr[v]) {
                    return false;
                }
            }
        }
        return true;
    }
};
MNNTestSuiteRegister(RasterOutputTest, "expr/RasterOutput");