From 29f9714e2aaab770232e4bc4cd0da762cbfc43ed Mon Sep 17 00:00:00 2001
From: Newnius <newnius.cn@gmail.com>
Date: Thu, 30 Apr 2020 12:02:56 +0800
Subject: [PATCH] update

---
 job.logic.php | 1 +
 modals.php    | 9 ++++++++-
 static/job.js | 2 ++
 3 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/job.logic.php b/job.logic.php
index cef30fe..488f672 100644
--- a/job.logic.php
+++ b/job.logic.php
@@ -31,6 +31,7 @@ function job_submit(CRObject $job)
 		$task['memory'] = intval($task['memory']);
 		$task['gpu_number'] = intval($task['gpu_number']);
 		$task['gpu_memory'] = intval($task['gpu_memory']);
+		$task['is_ps'] = $task['is_ps'] == 1;
 		$tasks[$i] = $task;
 	}
 	$job->set('tasks', $tasks);
diff --git a/modals.php b/modals.php
index bf3da9e..7e720da 100755
--- a/modals.php
+++ b/modals.php
@@ -155,13 +155,20 @@
 									<button type="button" class="btn btn-default task-remove">Remove</button>
 								</div>
 							</div>
-							<div class="col-md-2">
+							<div class="col-md-1">
 								<label>Host Name</label>
 								<div class="form-group">
 									<input type="text" class="form-control task-name" maxlength="32"
 									       placeholder="Task Name & Node Name" value="node1" required/>
 								</div>
 							</div>
+							<div class="col-md-1">
+								<label>Node Role<abbr title="Node role">?</abbr></label>
+								<select class="form-control form-control task-is-ps" required>
+									<option value="1">PS</option>
+									<option value="0" selected>Worker</option>
+								</select>
+							</div>
 							<div class="col-md-2">
 								<label>CPU Number</label>
 								<div class="form-group">
diff --git a/static/job.js b/static/job.js
index cf1c636..d969c94 100755
--- a/static/job.js
+++ b/static/job.js
@@ -57,6 +57,8 @@ function register_events_job() {
 			task['memory'] = $(this).find('.task-mem').eq(0).val();
 			task['gpu_number'] = $(this).find('.task-gpu-num').eq(0).val();
 			task['gpu_memory'] = $(this).find('.task-gpu-mem').eq(0).val();
+			task['is_ps'] = $(this).find('.task-is-ps').eq(0).val();
+			task['gpu_model'] = $(this).find('.task-gpu-model').eq(0).val();
 			tasks.push(task);
 		});