Corentin Salingue 12 роки тому
батько
коміт
5b1794cd20
1 змінених файлів з 38 додано та 8 видалено
  1. 38 8
      tests/disk/disk_compute.c

+ 38 - 8
tests/disk/disk_compute.c

@@ -21,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)
@@ -32,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;
 	
@@ -60,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;
 
@@ -72,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;
 
@@ -84,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;
@@ -107,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 */
@@ -127,9 +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();