wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-01 15:27:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Java] Add Plasma Free to Java code path (#2802)

Browse Source
This commit is contained in:
Yuhong Guo
2018-09-04 15:28:23 +08:00
committed by Hao Chen
parent 25ffe57a5c
commit dfb7c2be1e
9 changed files with 125 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
java/runtime-common/src/main/java/org/ray/core/AbstractRayRuntime.java
+5
View File
@@ -305,6 +305,11 @@ public abstract class AbstractRayRuntime implements RayRuntime {
}
}
@Override
public void free(List<UniqueId> objectIds, boolean localOnly) {
localSchedulerClient.freePlasmaObjects(objectIds, localOnly);
}
private List<List<UniqueId>> splitIntoBatches(List<UniqueId> objectIds, int batchSize) {
List<List<UniqueId>> batches = new ArrayList<>();
int objectsSize = objectIds.size();
java/runtime-common/src/main/java/org/ray/spi/LocalSchedulerLink.java
+2
View File
@@ -24,4 +24,6 @@ public interface LocalSchedulerLink {
UniqueId generateTaskId(UniqueId driverId, UniqueId parentTaskId, int taskIndex);
List<byte[]> wait(byte[][] objectIds, int timeoutMs, int numReturns);
void freePlasmaObjects(List<UniqueId> objectIds, boolean localOnly);
}