Maxime 4 lat temu
rodzic
commit
6b24f0f255
1 zmienionych plików z 6 dodań i 0 usunięć
  1. 6 0
      examples/mult/xgemm.c

+ 6 - 0
examples/mult/xgemm.c

@@ -407,6 +407,12 @@ static void parse_args(int argc, char **argv)
 				exit(EXIT_FAILURE);
 			}
 		}
+		
+		else if (strcmp(argv[i], "-xyz") == 0)
+		{
+			char *argptr;
+			xdim = ydim = zdim = strtol(argv[++i], &argptr, 10);
+		}
 
 		else if (strcmp(argv[i], "-y") == 0)
 		{