Function TYPED_TEST

test/reduce.cpp:625–641

}

TYPED_TEST(ReduceByKey, MultiBlockReduceSingleval) {
    SUPPORTED_TYPE_CHECK(TypeParam);
    array keys = constant(0, 1024 * 1024, s32);
    array vals = constant(1, 1024 * 1024,
                          (af_dtype)af::dtype_traits<TypeParam>::af_type);

    array keyResGold      = constant(0, 1);
    using promoted_t      = typename promote_type<TypeParam, af_sum>::type;
    array valsReducedGold = constant(
        1024 * 1024, 1, (af_dtype)af::dtype_traits<promoted_t>::af_type);

    array keyRes, valsReduced;
    sumByKey(keyRes, valsReduced, keys, vals);

    ASSERT_TRUE(allTrue<bool>(keyResGold == keyRes));
    ASSERT_ARRAYS_NEAR(valsReducedGold, valsReduced, 1e-5);
}

void reduce_by_key_test(std::string test_fn) {
    vector<dim4> numDims;