public:

virtual void SetUp() {

subMat0.push_back(af_make_seq(0, 4, 1));

subMat0.push_back(af_make_seq(0, 2, 1));

subMat0.push_back(af_make_seq(0, 1, 1));

subMat1.push_back(af_make_seq(1, 4, 1));

subMat1.push_back(af_make_seq(0, 2, 1));

subMat1.push_back(af_make_seq(0, 1, 1));

subMat2.push_back(af_make_seq(1, 4, 1));

subMat2.push_back(af_make_seq(0, 3, 1));

subMat2.push_back(af_make_seq(0, 2, 1));

}

vector<af_seq> subMat0;

vector<af_seq> subMat1;

vector<af_seq> subMat2;

const vector<af_seq> *seqv = NULL) {

SUPPORTED_TYPE_CHECK(T);

vector<dim4> numDims;

vector<vector<T>> in;

vector<vector<T>> tests;

readTests<T, T, int>(pTestFile, numDims, in, tests);

dim4 dims = numDims[0];

af_array inArray = 0;

af_array outArray = 0;

af_array tempArray = 0;

// Get input array

if (isSubRef) {

ASSERT_SUCCESS(af_create_array(&tempArray, &(in[0].front()),

dims.ndims(), dims.get(),

(af_dtype)dtype_traits<T>::af_type));

ASSERT_SUCCESS(

af_index(&inArray, tempArray, seqv->size(), &seqv->front()));

} else {

ASSERT_SUCCESS(af_create_array(&inArray, &(in[0].front()), dims.ndims(),

dims.get(),

(af_dtype)dtype_traits<T>::af_type));

}

// Run diff2

ASSERT_SUCCESS(af_diff2(&outArray, inArray, dim));

// Compare result

for (size_t testIter = 0; testIter < tests.size(); ++testIter) {

vector<T> currGoldBar = tests[testIter];

dim4 goldDims;

ASSERT_SUCCESS(af_get_dims(&goldDims[0], &goldDims[1], &goldDims[2],

&goldDims[3], inArray));

goldDims[dim] -= 2;

ASSERT_VEC_ARRAY_EQ(currGoldBar, goldDims, outArray);

}

if (inArray != 0) af_release_array(inArray);

if (outArray != 0) af_release_array(outArray);

if (tempArray != 0) af_release_array(tempArray);

SUPPORTED_TYPE_CHECK(T);

vector<dim4> numDims;

vector<vector<T>> in;

vector<vector<T>> tests;

readTests<T, T, int>(pTestFile, numDims, in, tests);

dim4 dims = numDims[0];

af_array inArray = 0;

af_array outArray = 0;

ASSERT_SUCCESS(af_create_array(&inArray, &(in[0].front()), dims.ndims(),

dims.get(),

(af_dtype)dtype_traits<T>::af_type));

ASSERT_EQ(AF_ERR_ARG, af_diff2(&outArray, inArray, -1));

ASSERT_EQ(AF_ERR_ARG, af_diff2(&outArray, inArray, 5));

if (inArray != 0) af_release_array(inArray);

if (outArray != 0) af_release_array(outArray);

const size_t largeDim = 65535 * 32 + 1;

deviceGC();

{

array in = constant(1, largeDim);

array diff = diff2(in, 0);

float s = sum<float>(diff, 1);

ASSERT_EQ(s, 0.f);

in = constant(1, 1, largeDim);

diff = diff2(in, 1);

s = sum<float>(diff, 1);

ASSERT_EQ(s, 0.f);

in = constant(1, 1, 1, largeDim);

diff = diff2(in, 2);

s = sum<float>(diff, 1);

ASSERT_EQ(s, 0.f);

in = constant(1, 1, 1, 1, largeDim);

diff = diff2(in, 3);

s = sum<float>(diff, 1);

ASSERT_EQ(s, 0.f);

}

const unsigned dim = 1;

vector<dim4> numDims;

vector<vector<float>> in;

vector<vector<float>> tests;

readTests<float, float, int>(string(TEST_DIR "/diff2/matrix1.test"),

numDims, in, tests);

dim4 dims = numDims[0];

array input(dims, &(in[0].front()));

array output = diff2(input, dim);

// Compare result

for (size_t testIter = 0; testIter < tests.size(); ++testIter) {

vector<float> currGoldBar = tests[testIter];

dim4 goldDims = input.dims();

goldDims[dim] -= 2;

ASSERT_VEC_ARRAY_EQ(currGoldBar, goldDims, output);

}