root/test/correctness/partition_loops_bug.cpp

DEFINITIONS

1. test
2. main

#include "Halide.h"

#include <iostream>

using namespace Halide;

Buffer<double> test(bool with_vectorize) {
    ImageParam input(Float(64), 2);

    Func output;

    Func input_padded = BoundaryConditions::constant_exterior(input, 100);

    RDom rk(-1,3,-1,3);

    Var x("x"), y("y");

    output(x,y) = sum(input_padded(x+rk.x,y+rk.y));

    if (with_vectorize) {
        output.vectorize(y,4);
    }

    Buffer<double> img = lambda(x, y, Expr(1.0)).realize(4, 4);
    input.set(img);

    Buffer<double> result(4, 4);

    input
        .dim(0).set_bounds(0, 4)
        .dim(1).set_bounds(0, 4);
    output.output_buffer()
        .dim(0).set_bounds(0, 4)
        .dim(1).set_bounds(0, 4);

    output.realize(result);

    return result;
}


int main (int argc, char const *argv[]) {

    if (sizeof(void *) == 4) {
        printf("Skipping this test. It triggers a bug in llvm 3.6.2 when run in 32-bit mode.\n");
        return 0;
        // The specific bug is:
        // void {anonymous}::SelectionDAGLegalize::LegalizeOp(llvm::SDNode*): Assertion `(TLI.getTypeAction(*DAG.getContext(), Node->getOperand(i).getValueType()) == TargetLowering::TypeLegal || Node->getOperand(i).getOpcode() == ISD::TargetConstant) && "Unexpected illegal type!"' failed.
    }

    Buffer<double> im1 = test(true);
    Buffer<double> im2 = test(false);

    for (int y = 0; y < im1.height(); y++) {
        for (int x = 0; x < im1.width(); x++) {
            if (im1(x, y) != im2(x, y)) {
                printf("im1(%d, %d) = %f, im2(%d, %d) = %f\n",
                       x, y, im1(x, y),
                       x, y, im2(x, y));
                return -1;
            }
        }
    }

    printf("Success!\n");

    return 0;
}