wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-12 11:59:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

[xray] Track ray.get calls as task dependencies (#2362)

Browse Source
This commit is contained in:
Stephanie Wang
2018-07-27 14:59:17 -04:00
committed by Robert Nishihara
parent 5b015f9a79
commit 6675361684
20 changed files with 472 additions and 198 deletions
Download Patch File Download Diff File Expand all files Collapse all files
java/runtime-common/src/main/java/org/ray/core/WorkerContext.java
+6 -1
View File
@@ -2,6 +2,7 @@ package org.ray.core;
import org.ray.api.UniqueID;
import org.ray.core.model.RayParameters;
import org.ray.core.model.WorkerMode;
import org.ray.spi.model.TaskSpec;
public class WorkerContext {
@@ -35,7 +36,11 @@ public class WorkerContext {
TaskSpec dummy = new TaskSpec();
dummy.parentTaskId = UniqueID.nil;
dummy.taskId = UniqueID.nil;
if (params.worker_mode == WorkerMode.DRIVER) {
dummy.taskId = UniqueID.randomId();
} else {
dummy.taskId = UniqueID.nil;
}
dummy.actorId = UniqueID.nil;
dummy.driverId = params.driver_id;
prepare(dummy, null);