SUPPORTED_TYPE_CHECK(af_half);

array aa = af::constant(3.14, 3, 3, f16);

array bb = af::constant(2, 3, 3, f16);

af_print(aa);

af_dtype from, to;

double value;

convert_params(af_dtype f, af_dtype t, double v)

: from(f), to(t), value(v) {}

SUPPORTED_TYPE_CHECK(af_half);

convert_params params = GetParam();

if (noDoubleTests(params.to))

GTEST_SKIP() << "Double not supported on this device";

if (noDoubleTests(params.from))

GTEST_SKIP() << "Double not supported on this device";

array from = af::constant(params.value, 3, 3, params.from);

array to = from.as(params.to);

ASSERT_EQ(from.type(), params.from);

ASSERT_EQ(to.type(), params.to);

array gold = af::constant(params.value, 3, 3, params.to);

ASSERT_ARRAYS_EQ(gold, to);

SUPPORTED_TYPE_CHECK(af_half);

array aa = af::constant(3.14, 3, 3, f16);

array bb = af::constant(1, 3, 3, f16);

array gold = constant(4.14, 3, 3, f16);

array result = bb + aa;

ASSERT_ARRAYS_EQ(gold, result);

SUPPORTED_TYPE_CHECK(af_half);

SKIP_IF_FAST_MATH_ENABLED();

half_float::half hinf = std::numeric_limits<half_float::half>::infinity();

vector<half_float::half> input(2, half_float::half(0));

input[0] = hinf;

array infarr(2, &input.front());

array res = isInf(infarr);

vector<char> hgold(2, 0);

hgold[0] = 1;

array gold(2, &hgold.front());

ASSERT_ARRAYS_EQ(gold, res);

SUPPORTED_TYPE_CHECK(af_half);

SKIP_IF_FAST_MATH_ENABLED();

half_float::half hnan = std::numeric_limits<half_float::half>::quiet_NaN();

vector<half_float::half> input(2, half_float::half(0));

input[0] = hnan;

array nanarr(2, &input.front());

array res = isNaN(nanarr);

vector<char> hgold(2, 0);

hgold[0] = 1;

array gold(2, &hgold.front());

ASSERT_ARRAYS_EQ(gold, res);

SUPPORTED_TYPE_CHECK(af_half);

half_float::half hzero(0.f);

vector<half_float::half> input(2, half_float::half(1));

input[0] = hzero;

array nanarr(2, &input.front());

array res = iszero(nanarr);

vector<char> hgold(2, 0);

hgold[0] = 1;

array gold(2, &hgold.front());

ASSERT_ARRAYS_EQ(gold, res);