Преглед изворни кода

Use by default bigger matrices, but less iterations

Samuel Thibault пре 12 година
родитељ
комит
9459043d04
1 измењених фајлова са 22 додато и 9 уклоњено
  1. 22 9
      examples/mult/xgemm.c

+ 22 - 9
examples/mult/xgemm.c

@@ -30,12 +30,18 @@
 #include <cublas.h>
 #endif
 
-static unsigned niter = 100;
+static unsigned niter = 10;
 static unsigned nslicesx = 4;
 static unsigned nslicesy = 4;
+#ifdef STARPU_SLOW_MACHINE
 static unsigned xdim = 256;
 static unsigned ydim = 256;
 static unsigned zdim = 64;
+#else
+static unsigned xdim = 1024;
+static unsigned ydim = 1024;
+static unsigned zdim = 1024;
+#endif
 static unsigned check = 0;
 
 static TYPE *A, *B, *C;
@@ -218,51 +224,58 @@ static void parse_args(int argc, char **argv)
 			nslicesy = nslicesx;
 		}
 
-		if (strcmp(argv[i], "-nblocksx") == 0)
+		else if (strcmp(argv[i], "-nblocksx") == 0)
 		{
 			char *argptr;
 			nslicesx = strtol(argv[++i], &argptr, 10);
 		}
 
-		if (strcmp(argv[i], "-nblocksy") == 0)
+		else if (strcmp(argv[i], "-nblocksy") == 0)
 		{
 			char *argptr;
 			nslicesy = strtol(argv[++i], &argptr, 10);
 		}
 
-		if (strcmp(argv[i], "-x") == 0)
+		else if (strcmp(argv[i], "-x") == 0)
 		{
 			char *argptr;
 			xdim = strtol(argv[++i], &argptr, 10);
 		}
 
-		if (strcmp(argv[i], "-y") == 0)
+		else if (strcmp(argv[i], "-y") == 0)
 		{
 			char *argptr;
 			ydim = strtol(argv[++i], &argptr, 10);
 		}
 
-		if (strcmp(argv[i], "-z") == 0)
+		else if (strcmp(argv[i], "-z") == 0)
 		{
 			char *argptr;
 			zdim = strtol(argv[++i], &argptr, 10);
 		}
 
-		if (strcmp(argv[i], "-iter") == 0)
+		else if (strcmp(argv[i], "-iter") == 0)
 		{
 			char *argptr;
 			niter = strtol(argv[++i], &argptr, 10);
 		}
 
-		if (strcmp(argv[i], "-check") == 0)
+		else if (strcmp(argv[i], "-check") == 0)
 		{
 			check = 1;
 		}
 
-		if (strcmp(argv[i], "-spmd") == 0)
+		else if (strcmp(argv[i], "-spmd") == 0)
 		{
 			cl.type = STARPU_SPMD;
 		}
+
+		else if (strcmp(argv[i], "-help") == 0 || strcmp(argv[i], "--help") == 0 || strcmp(argv[i], "-h") == 0)
+		{
+			fprintf(stderr,"Usage: %s [-nblocks n] [-nblocksx x] [-nblocksy y] [-x x] [-y y] [-z z] [-iter iter] [-check] [-spmd]\n", argv[0]);
+			fprintf(stderr,"Currently selected: %ux%u * %ux%u and %ux%u blocks, %u iterations\n", zdim, ydim, xdim, zdim, nslicesx, nslicesy, niter);
+			exit(EXIT_SUCCESS);
+		}
 	}
 }