Function TEST_P

test/binary.cpp:676–692

}

TEST_P(Broadcast, Addition) {
    auto params   = GetParam();
    af::array lhs = iota(get<0>(params));
    af::array rhs = constant(1, get<1>(params));

    af::array out = lhs + rhs;

    af::dim4 outdims       = broadcastOut(lhs.dims(), rhs.dims());
    af::dim4 tilerepetions = tileRepeations(lhs.dims(), rhs.dims());
    af::array tiledlhs     = tile(lhs, tilerepetions);

    vector<float> outvec(outdims.elements());
    tiledlhs.host(outvec.data());
    for (auto &out : outvec) { out += 1; }

    ASSERT_VEC_ARRAY_EQ(outvec, outdims, out);
}

TEST_P(Broadcast, Subtraction) {
    auto params   = GetParam();