Function argsort_data_gen

src/opr/test/misc.cpp:26–42

}

void argsort_data_gen(HostTensorND& dest) {
    mgb_assert(dest.layout().ndim == 2 && dest.layout().is_contiguous());
    size_t m = dest.layout()[0], n = dest.layout()[1];
    auto ptr = dest.ptr<float>();
    RNGxorshf rng{next_rand_seed()};
    std::uniform_real_distribution<float> dist_base{-10.f, 10.f},
            dist_delta{0.1f, 1.2f};
    for (size_t i = 0; i < m; ++i) {
        auto v = dist_base(rng);
        for (size_t j = 0; j < n; ++j) {
            ptr[j] = v;
            v += dist_delta(rng);
        }
        std::shuffle(ptr, ptr + n, rng);
        ptr += n;
    }
}
}  // namespace

TEST(TestOprMisc, Argmxx) {