From a50df2b46f821404e216f935effda5dcae390e72 Mon Sep 17 00:00:00 2001
From: Newnius <newnius.cn@gmail.com>
Date: Mon, 4 May 2020 12:14:01 +0800
Subject: [PATCH] update

---
 src/job_manager.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/job_manager.go b/src/job_manager.go
index 15cd925..f5c53f7 100644
--- a/src/job_manager.go
+++ b/src/job_manager.go
@@ -115,7 +115,9 @@ func (jm *JobManager) start() {
 			v.Set("cpu_limit", strconv.Itoa(jm.job.Tasks[i].NumberCPU))
 			v.Set("network", network)
 			v.Set("should_wait", "1")
-			v.Set("HDFS_path", "1")
+			v.Set("output_dir", "/output/")
+			v.Set("hdfs_dir", "http://hdfs-master:50070/user/yao/output/"+jm.job.Name)
+			v.Set("gpu_mem", strconv.Itoa(jm.job.Tasks[i].MemoryGPU))
 
 			resp, err := doRequest("POST", "http://"+jm.resources[i].ClientHost+":8000/create", strings.NewReader(v.Encode()), "application/x-www-form-urlencoded", "")
 			if err != nil {