Method SetUp

test/reduce.cpp:2019–2040

    array valsReducedGold, idxsReducedGold;

    void SetUp() {
        ragged_params *params = GetParam();
        if (noHalfTests(params->vType_)) {
            GTEST_SKIP() << "Half not supported on this device";
        }
        if (noDoubleTests(GetParam()->vType_)) {
            GTEST_SKIP() << "Double not supported on this device";
        }

        const size_t rdim_size = params->reduceDimLen_;
        const int dim          = params->reduceDim_;

        af::dim4 rdim(3, 3, 3, 3);
        rdim[dim] = rdim_size;
        vals      = af::range(rdim, dim, params->vType_);

        rdim[dim]   = 1;
        ragged_lens = af::range(rdim, (dim > 0) ? 0 : 1, params->lType_) + 1;

        valsReducedGold = af::range(rdim, (dim > 0) ? 0 : 1, params->oType_);
        idxsReducedGold = af::range(rdim, (dim > 0) ? 0 : 1, params->lType_);
    }

    void TearDown() { delete GetParam(); }
};