diff TaskManager/kernel/main.cc @ 1626:4fed76f4d101 draft

add gpuNum
author shuhei kokubo
date Thu, 30 May 2013 18:35:08 +0900
parents 4443d59a8210
children a11565fea1e5
line wrap: on
line diff
--- a/TaskManager/kernel/main.cc	Wed May 29 21:42:47 2013 +0900
+++ b/TaskManager/kernel/main.cc	Thu May 30 18:35:08 2013 +0900
@@ -35,12 +35,15 @@
 main(int argc, char *argv[])
 {
     int machineNum = 1;
+    int gpuNum = 0;
     int speIdle = 0;
     int useRefDma = 0;
 
     for (int i = 1; argv[i]; ++i) {
         if (strcmp(argv[i], "-cpu") == 0) {
             machineNum = atoi(argv[++i]);
+	} else if (strcmp(argv[i], "-gpu") == 0) {
+            gpuNum = atoi(argv[++i]);
         } else if (strcmp(argv[i], "-p") == 0) {
             profile = 1;
         } else if (strcmp(argv[i], "-export") == 0) {
@@ -54,7 +57,7 @@
         }
     }
 
-    TaskManager *manager = new TaskManager(machineNum);
+    TaskManager *manager = new TaskManager(machineNum, gpuNum);
     manager->init(speIdle,export_task_log, useRefDma);
 
     manager->set_TMend(defaultTMend);