megenginelite-sys 1.8.2

A safe megenginelite wrapper in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

/**
 * \file src/opr/test/dnn/roi_align.cpp
 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
 *
 * Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 */

#include "megbrain/opr/dnn/roi_align.h"
#include "megbrain/test/autocheck.h"
#include "megbrain/test/helper.h"
#include "megbrain/test/megdnn_helper.h"

#include "megdnn/oprs.h"

#include <cmath>
#include <iomanip>
#include <random>
#include <sstream>

using namespace mgb;

namespace {
using Param = opr::ROIAlignForward::Param;

void run(Param::Mode mode) {
    RNGxorshf rng{next_rand_seed()};

    using Checker = AutoOprChecker<2, 2>;

    size_t N = 2, C = 3, M = 4;

    TensorShape rois_shp{M, 5};

    Param param;
    param.mode = mode;
    param.pooled_height = 2;
    param.pooled_width = 3;
    param.sample_height = 5;
    param.sample_width = 6;

    TensorShape dst_shp{M, C, param.pooled_height, param.pooled_width};
    auto make_graph = [&](const Checker::SymInpArray& inputs) -> Checker::SymOutArray {
        auto o0 = opr::ROIAlignForward::make(inputs[0], inputs[1], param);
        return {o0, o0.node()->owner_opr()->output(1)};
    };

    auto fwd = [&](Checker::NumOutArray& dest, Checker::NumInpArray inp) {
        auto opr = megdnn_naive_handle()->create_operator<megdnn::ROIAlign>();
        opr->param() = param;
        dest[0].dtype(dtype::Float32()).comp_node(inp[0]->comp_node()).resize(dst_shp);
        dest[1].dtype(dtype::Int32()).comp_node(inp[0]->comp_node()).resize(dst_shp);
        opr->exec(
                inp[0]->as_megdnn(), inp[1]->as_megdnn(), dest[0].as_megdnn(),
                dest[1].as_megdnn(), {});
    };
    auto rand_int = [&](int lo, int hi) {
        std::uniform_int_distribution<int> dist(lo, hi);
        return dist(rng);
    };
    auto rand_real = [&](float lo, float hi) {
        std::uniform_real_distribution<float> dist(lo, hi);
        return dist(rng);
    };
    auto gen_rois = [&](HostTensorND& rois) {
        auto ptr = rois.ptr<float>();
        for (size_t i = 0; i < M; ++i) {
            ptr[0] = rand_int(0, N - 1);
            ptr[1] = rand_real(0.0f, 1.0f);
            ptr[2] = rand_real(0.0f, 1.0f);
            ptr[3] = rand_real(0.0f, 1.0f);
            ptr[4] = rand_real(0.0f, 1.0f);
            if (ptr[1] > ptr[3])
                std::swap(ptr[1], ptr[3]);
            if (ptr[2] > ptr[4])
                std::swap(ptr[2], ptr[4]);
            ptr += 5;
        };
        mgb_assert(ptr == rois.ptr<float>() + rois.shape().total_nr_elems());
    };
    Checker::RunOptions option;
    option.numdiff_eps = 1e-3;
    option.numdiff_max_err = 1e-2;
    Checker checker{make_graph, fwd};

    checker.set_input_generator(1, gen_rois)
            .
            // we cannot take gradient wrt. rois
            set_input_allow_grad(1, false)
            .
            // we cannot take gradient wrt. temporary output
            set_output_allow_grad(1, false);

    if (mode == Param::Mode::AVERAGE)
        checker.set_output_allow_check(1, false);

    checker.run({TensorShape{M, C, 5, 6}, rois_shp}, option)
            .run({TensorShape{M, C, 7, 8}, rois_shp}, option)
            .run({TensorShape{M, C, 4, 2}, rois_shp}, option);
}
}  // anonymous namespace

TEST(TestOprDNN, ROIAlignMax) {
    // TODO: fix me, add correct backward of cpu
    REQUIRE_GPU(1);
    run(Param::Mode::MAX);
}

TEST(TestOprDNN, ROIAlignAverage) {
    // TODO: fix me, add correct backward of cpu
    REQUIRE_GPU(1);
    run(Param::Mode::AVERAGE);
}

// vim: syntax=cpp.doxygen foldmethod=marker foldmarker=f{{{,f}}}