[core] Pass owner address from the workers to the raylet (#9299)

* Add intended worker ID to GetObjectStatus, tests

* Remove TaskID owner_id

* lint

* Add owner address to task args

* Make TaskArg a virtual class, remove multi args

* Set owner address for task args

* merge

* Fix tests

* Add ObjectRefs to task dependency manager, pass from task spec args

* tmp

* tmp

* Fix

* Add ownership info for task arguments

* Convert WaitForDirectActorCallArgs

* lint

* build

* update

* build

* java

* Move code

* build

* Revert "Fix Google log directory again (#9063)"

This reverts commit 275da2e400.

* Fix free

* fix tests

* Fix tests

* build

* build

* fix

* Change assertion to warning to fix java
This commit is contained in:
Stephanie Wang
2020-07-09 14:35:54 -07:00
committed by GitHub
parent 4687b807c4
commit 0389735d7a
28 changed files with 409 additions and 199 deletions
+5 -3
View File
@@ -38,10 +38,12 @@ def _fill_object_store_and_get(oid, succeed=True, object_MiB=40,
oid = ray.ObjectID(oid)
if succeed:
ray.get(oid)
wait_for_condition(
lambda: ray.worker.global_worker.core_worker.object_exists(oid))
else:
with pytest.raises(ray.exceptions.RayTimeoutError):
ray.get(oid, timeout=0.1)
wait_for_condition(
lambda: not ray.worker.global_worker.core_worker.object_exists(oid)
)
def _check_refcounts(expected):
@@ -41,10 +41,12 @@ def _fill_object_store_and_get(oid, succeed=True, object_MiB=40,
oid = ray.ObjectID(oid)
if succeed:
ray.get(oid)
wait_for_condition(
lambda: ray.worker.global_worker.core_worker.object_exists(oid))
else:
with pytest.raises(ray.exceptions.RayTimeoutError):
ray.get(oid, timeout=0.1)
wait_for_condition(
lambda: not ray.worker.global_worker.core_worker.object_exists(oid)
)
# Test that an object containing object IDs within it pins the inner IDs