Pārlūkot izejas kodu

tests: check functionality introduced in #5626

Nathalie Furmento 13 gadi atpakaļ
vecāks
revīzija
9cf56e891a
2 mainītis faili ar 107 papildinājumiem un 0 dzēšanām
  1. 1 0
      tests/Makefile.am
  2. 106 0
      tests/datawizard/interfaces/copy_interfaces.c

+ 1 - 0
tests/Makefile.am

@@ -169,6 +169,7 @@ noinst_PROGRAMS =				\
 	datawizard/increment_redux_lazy		\
 	datawizard/handle_to_pointer		\
 	datawizard/lazy_allocation		\
+	datawizard/interfaces/copy_interfaces	\
 	datawizard/interfaces/block/block_interface \
 	datawizard/interfaces/bcsr/bcsr_interface \
 	datawizard/interfaces/csr/csr_interface \

+ 106 - 0
tests/datawizard/interfaces/copy_interfaces.c

@@ -0,0 +1,106 @@
+/* StarPU --- Runtime system for heterogeneous multicore architectures.
+ *
+ * Copyright (C) 2012  Centre National de la Recherche Scientifique
+ *
+ * StarPU is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation; either version 2.1 of the License, or (at
+ * your option) any later version.
+ *
+ * StarPU is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
+ *
+ * See the GNU Lesser General Public License in COPYING.LGPL for more details.
+ */
+
+#include <starpu.h>
+#include "../../helper.h"
+#include <datawizard/coherency.h>
+
+static int check_copy(starpu_data_handle_t handle, char *header)
+{
+	void *old_interface, *new_interface;
+	starpu_data_handle_t new_handle;
+	int ret=0;
+
+	STARPU_ASSERT(handle->ops->allocate_new_data);
+	handle->ops->allocate_new_data(handle, &new_interface);
+	starpu_data_register(&new_handle, -1, new_interface, handle->ops);
+
+	if (!getenv("STARPU_SSILENT") && new_handle->ops->display)
+	{
+		fprintf(stderr, "%s: ", header);
+		new_handle->ops->display(new_handle, stderr);
+		fprintf(stderr, "\n");
+	}
+
+	old_interface = starpu_data_get_interface_on_node(handle, 0);
+	if (new_handle->ops->compare(old_interface, new_interface) == 0)
+	{
+		FPRINTF(stderr, "Error when copying %s data\n", header);
+		assert(0);
+		ret = 1;
+	}
+	starpu_data_unregister(handle);
+	starpu_data_unregister(new_handle);
+	return ret;
+}
+
+int main(int argc, char **argv)
+{
+	int ret;
+	starpu_data_handle_t handle;
+
+	ret = starpu_init(NULL);
+	if (ret == -ENODEV) return STARPU_TEST_SKIPPED;
+	STARPU_CHECK_RETURN_VALUE(ret, "starpu_init");
+
+	{
+		int x=42;
+		starpu_variable_data_register(&handle, 0, (uintptr_t)&x, sizeof(x));
+		ret = check_copy(handle, "variable");
+	}
+
+	if (ret == 0)
+	{
+		int xx[] = {12, 23, 45};
+		starpu_vector_data_register(&handle, 0, (uintptr_t)xx, 3, sizeof(xx[0]));
+		ret = check_copy(handle, "vector");
+	}
+
+	if (ret == 0)
+	{
+		int NX=3;
+		int NY=2;
+		int matrix[NX][NY];
+		starpu_matrix_data_register(&handle, 0, (uintptr_t)matrix, NX, NX, NY, sizeof(matrix[0]));
+		ret = check_copy(handle, "matrix");
+	}
+
+	if (ret == 0)
+	{
+		int NX=3;
+		int NY=2;
+		int NZ=4;
+		int block[NX*NY*NZ];
+		starpu_block_data_register(&handle, 0, (uintptr_t)block, NX, NX*NY, NX, NY, NZ, sizeof(block[0]));
+		ret = check_copy(handle, "block");
+	}
+
+	if (ret == 0)
+	{
+		uint32_t nnz = 2;
+		unsigned nrow = 5;
+		float nzvalA[20];
+		uint32_t colind[1];
+		uint32_t rowptr[2];
+		starpu_csr_data_register(&handle, 0, nnz, nrow, (uintptr_t)nzvalA, colind, rowptr, 0, sizeof(float));
+		ret = check_copy(handle, "csr");
+	}
+
+	starpu_shutdown();
+	return ret;
+}
+
+