|
@@ -0,0 +1,179 @@
|
|
|
+/* StarPU --- Runtime system for heterogeneous multicore architectures.
|
|
|
+ *
|
|
|
+ * Copyright (C) 2013 Corentin Salingue
|
|
|
+ *
|
|
|
+ * 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.
|
|
|
+ */
|
|
|
+//! [To be included]
|
|
|
+/* Try to write into disk memory
|
|
|
+ * Use mechanism to push datas from main ram to disk ram
|
|
|
+ */
|
|
|
+
|
|
|
+#include <starpu.h>
|
|
|
+#include <stdlib.h>
|
|
|
+#include <stdio.h>
|
|
|
+#include <sys/types.h>
|
|
|
+#include <unistd.h>
|
|
|
+#include <math.h>
|
|
|
+
|
|
|
+#define NX (100)
|
|
|
+
|
|
|
+int main(int argc, char **argv)
|
|
|
+{
|
|
|
+ /* Initialize StarPU with default configuration */
|
|
|
+ int ret = starpu_init(NULL);
|
|
|
+
|
|
|
+ if (ret == -ENODEV) goto enodev;
|
|
|
+
|
|
|
+ /* Initialize path and name */
|
|
|
+ char pid_str[16];
|
|
|
+ int pid = getpid();
|
|
|
+ snprintf(pid_str, 16, "%d", pid);
|
|
|
+
|
|
|
+ char * base = "/tmp/";
|
|
|
+
|
|
|
+ char * name_file_start = malloc(128*sizeof(char));
|
|
|
+ strcpy(name_file_start, "STARPU_DISK_COMPUTE_DATA_");
|
|
|
+ strcat(name_file_start, pid_str);
|
|
|
+
|
|
|
+ char * name_file_end = malloc(128*sizeof(char));
|
|
|
+ strcpy(name_file_end, "STARPU_DISK_COMPUTE_DATA_RESULT_");
|
|
|
+ strcat(name_file_end, pid_str);
|
|
|
+
|
|
|
+ char * path_file_start = malloc(128*sizeof(char));
|
|
|
+ strcpy(path_file_start, base);
|
|
|
+ strcat(path_file_start, name_file_start);
|
|
|
+
|
|
|
+ char * path_file_end = malloc(128*sizeof(char));
|
|
|
+ strcpy(path_file_end, base);
|
|
|
+ strcat(path_file_end, name_file_end);
|
|
|
+
|
|
|
+
|
|
|
+ /* register a disk */
|
|
|
+ int new_dd = starpu_disk_register(&starpu_disk_stdio_ops, (void *) base, 1024*1024*1);
|
|
|
+ /* can't write on /tmp/ */
|
|
|
+ if (new_dd == -ENOENT) goto enoent;
|
|
|
+
|
|
|
+ unsigned dd = (unsigned) new_dd;
|
|
|
+
|
|
|
+ printf("TEST DISK MEMORY \n");
|
|
|
+
|
|
|
+ /* Imagine, you want to compute datas */
|
|
|
+ int *A;
|
|
|
+ int *C;
|
|
|
+
|
|
|
+ starpu_malloc_flags((void **)&A, NX*sizeof(int), STARPU_MALLOC_COUNT);
|
|
|
+ starpu_malloc_flags((void **)&C, NX*sizeof(int), STARPU_MALLOC_COUNT);
|
|
|
+
|
|
|
+ unsigned int j;
|
|
|
+ /* you register them in a vector */
|
|
|
+ for(j = 0; j < NX; ++j)
|
|
|
+ {
|
|
|
+ A[j] = j;
|
|
|
+ C[j] = 0;
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ /* you create a file to store the vector ON the disk */
|
|
|
+ FILE * f = fopen(path_file_start, "wb+");
|
|
|
+ if (f == NULL)
|
|
|
+ goto enoent;
|
|
|
+
|
|
|
+ /* store it in the file */
|
|
|
+ fwrite(A, sizeof(int), NX, f);
|
|
|
+
|
|
|
+ /* close the file */
|
|
|
+ fclose(f);
|
|
|
+
|
|
|
+
|
|
|
+ /* create a file to store result */
|
|
|
+ f = fopen(path_file_end, "wb+");
|
|
|
+ if (f == NULL)
|
|
|
+ goto enoent;
|
|
|
+
|
|
|
+ /* replace all datas by 0 */
|
|
|
+ fwrite(C, sizeof(int), NX, f);
|
|
|
+
|
|
|
+ /* close the file */
|
|
|
+ fclose(f);
|
|
|
+
|
|
|
+ /* And now, you want to use your datas in StarPU */
|
|
|
+ /* Open the file ON the disk */
|
|
|
+ void * data = starpu_disk_open(dd, (void *) name_file_start, NX*sizeof(int));
|
|
|
+ void * data_result = starpu_disk_open(dd, (void *) name_file_end, NX*sizeof(int));
|
|
|
+
|
|
|
+
|
|
|
+ starpu_data_handle_t vector_handleA, vector_handleC;
|
|
|
+
|
|
|
+ /* register vector in starpu */
|
|
|
+ starpu_vector_data_register(&vector_handleA, dd, (uintptr_t) data, NX, sizeof(int));
|
|
|
+
|
|
|
+ /* and do what you want with it, here we copy it into an other vector */
|
|
|
+ starpu_vector_data_register(&vector_handleC, dd, (uintptr_t) data_result, NX, sizeof(int));
|
|
|
+
|
|
|
+ starpu_data_cpy(vector_handleC, vector_handleA, 0, NULL, NULL);
|
|
|
+
|
|
|
+ /* free them */
|
|
|
+ starpu_data_unregister(vector_handleA);
|
|
|
+ starpu_data_unregister(vector_handleC);
|
|
|
+
|
|
|
+ /* close them in StarPU */
|
|
|
+ starpu_disk_close(dd, data, NX*sizeof(int));
|
|
|
+ starpu_disk_close(dd, data_result, NX*sizeof(int));
|
|
|
+
|
|
|
+ /* check results */
|
|
|
+ f = fopen(path_file_end, "rb+");
|
|
|
+ if (f == NULL)
|
|
|
+ goto enoent;
|
|
|
+ /* take datas */
|
|
|
+ int size = fread(C, sizeof(int), NX, f);
|
|
|
+
|
|
|
+ /* close the file */
|
|
|
+ fclose(f);
|
|
|
+
|
|
|
+ int try = 1;
|
|
|
+ for (j = 0; j < NX; ++j)
|
|
|
+ if (A[j] != C[j])
|
|
|
+ {
|
|
|
+ printf("Fail A %d != C %d \n", A[j], C[j]);
|
|
|
+ try = 0;
|
|
|
+ }
|
|
|
+
|
|
|
+ starpu_free_flags(A, NX*sizeof(double), STARPU_MALLOC_COUNT);
|
|
|
+ starpu_free_flags(C, NX*sizeof(double), STARPU_MALLOC_COUNT);
|
|
|
+
|
|
|
+ unlink(path_file_start);
|
|
|
+ unlink(path_file_end);
|
|
|
+
|
|
|
+ free(name_file_start);
|
|
|
+ free(name_file_end);
|
|
|
+ free(path_file_start);
|
|
|
+ free(path_file_end);
|
|
|
+
|
|
|
+ /* terminate StarPU, no task can be submitted after */
|
|
|
+ starpu_shutdown();
|
|
|
+
|
|
|
+ if(try)
|
|
|
+ printf("TEST SUCCESS\n");
|
|
|
+ else
|
|
|
+ printf("TEST FAIL\n");
|
|
|
+ return (try ? EXIT_SUCCESS : EXIT_FAILURE);
|
|
|
+
|
|
|
+enodev:
|
|
|
+ return 77;
|
|
|
+enoent:
|
|
|
+ return 77;
|
|
|
+}
|
|
|
+//! [To be included]
|
|
|
+
|