Browse Source

update test in doc

Corentin Salingue 12 years ago
parent
commit
19d955bb82
1 changed files with 40 additions and 10 deletions
  1. 40 10
      doc/doxygen/chapters/code/disk_compute.c

+ 40 - 10
doc/doxygen/chapters/code/disk_compute.c

@@ -14,7 +14,6 @@
  * 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
  */
@@ -22,6 +21,8 @@
 #include <starpu.h>
 #include <stdlib.h>
 #include <stdio.h>
+#include <sys/types.h>
+#include <unistd.h>
 #include <math.h>
 
 #define NX (100)
@@ -33,8 +34,32 @@ int main(int argc, char **argv)
 
 	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 *) "/tmp/", 1024*1024*1);
+	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;
 	
@@ -61,7 +86,7 @@ int main(int argc, char **argv)
 
 
 	/* you create a file to store the vector ON the disk */
-	FILE * f = fopen("/tmp/STARPU_DISK_COMPUTE_DATA", "wb+");
+	FILE * f = fopen(path_file_start, "wb+");
 	if (f == NULL)
 		goto enoent;
 
@@ -73,7 +98,7 @@ int main(int argc, char **argv)
 
 
 	/* create a file to store result */
-	f = fopen("/tmp/STARPU_DISK_COMPUTE_DATA_RESULT", "wb+");
+	f = fopen(path_file_end, "wb+");
 	if (f == NULL)
 		goto enoent;
 
@@ -85,8 +110,8 @@ int main(int argc, char **argv)
 
 	/* And now, you want to use your datas in StarPU */
 	/* Open the file ON the disk */
-	void * data = starpu_disk_open(dd, (void *) "STARPU_DISK_COMPUTE_DATA", NX*sizeof(int));
-	void * data_result = starpu_disk_open(dd, (void *) "STARPU_DISK_COMPUTE_DATA_RESULT", NX*sizeof(int));
+	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;
@@ -108,7 +133,7 @@ int main(int argc, char **argv)
 	starpu_disk_close(dd, data_result, NX*sizeof(int));
 
 	/* check results */	
-	f = fopen("/tmp/STARPU_DISK_COMPUTE_DATA_RESULT", "rb+");
+	f = fopen(path_file_end, "rb+");
 	if (f == NULL)
 		goto enoent;
 	/* take datas */
@@ -128,8 +153,13 @@ int main(int argc, char **argv)
 	starpu_free_flags(A, NX*sizeof(double), STARPU_MALLOC_COUNT);
 	starpu_free_flags(C, NX*sizeof(double), STARPU_MALLOC_COUNT);
 
-	unlink("/tmp/STARPU_DISK_COMPUTE_DATA");
-	unlink("/tmp/STARPU_DISK_COMPUTE_DATA_RESULT");
+	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();
@@ -145,5 +175,5 @@ enodev:
 enoent:
 	return 77;
 }
-
 //! [To be included]
+