root/test/correctness/float16_t_image_type.cpp

DEFINITIONS

1. h_assert
2. main

#include "Halide.h"
#include <stdio.h>
#include <cmath>

using namespace Halide;

// FIXME: Why aren't we using a unit test framework for this?
void h_assert(bool condition, const char* msg) {
    if (!condition) {
        printf("FAIL: %s\n", msg);
        abort();
    }
}

int main() {
    Halide::Buffer<float16_t> simple(/*x=*/10, /*y=*/3);

    h_assert(sizeof(float16_t) == 2, "float16_t has invalid size");

    // Assert some basic properties of the image
    h_assert(simple.extent(0) == 10, "invalid width");
    h_assert(simple.extent(1) == 3, "invalid height");
    h_assert(simple.min(0) == 0, "unexpected non zero min in x");
    h_assert(simple.min(1) == 0, "unexpected non zero min in y");
    h_assert(simple.channels() == 1, "invalid channels");
    h_assert(simple.dimensions() == 2, "invalid number of dimensions");

    const float16_t zeroPointTwoFive = float16_t("0.25", RoundingMode::ToNearestTiesToEven);

    // Write a constant value and check we can read it back
    for (int x=simple.min(0); x < simple.extent(0); ++x) {
        for (int y = simple.min(1); y < simple.extent(1); ++y) {
            simple(x, y) = zeroPointTwoFive;
        }
    }

    for (int x = simple.min(0); x < simple.extent(0); ++x) {
        for (int y = simple.min(1); y < simple.extent(1); ++y) {
            h_assert(simple(x, y) == zeroPointTwoFive, "Invalid value read back");
            h_assert(simple(x, y).to_bits() == 0x3400, "Bit pattern incorrect");
        }
    }

    printf("Success!\n");
    return 0;
}