mirror of
https://github.com/wassname/ray.git
synced 2026-06-30 10:33:24 +08:00
Eager eviction even when object pinning is disabled (#8561)
* Eager eviction even when object pinning is disabled, add regression test * Make test more robust * lint
This commit is contained in:
@@ -917,12 +917,27 @@ def test_fill_object_store_exception(shutdown_only):
|
||||
|
||||
|
||||
def test_fill_object_store_lru_fallback(shutdown_only):
|
||||
ray.init(num_cpus=2, object_store_memory=10**8, lru_evict=True)
|
||||
config = json.dumps({
|
||||
"free_objects_batch_size": 1,
|
||||
})
|
||||
ray.init(
|
||||
num_cpus=2,
|
||||
object_store_memory=10**8,
|
||||
lru_evict=True,
|
||||
_internal_config=config)
|
||||
|
||||
@ray.remote
|
||||
def expensive_task():
|
||||
return np.zeros((10**8) // 2, dtype=np.uint8)
|
||||
|
||||
# Check that objects out of scope are cleaned up quickly.
|
||||
ray.get(expensive_task.remote())
|
||||
start = time.time()
|
||||
for _ in range(3):
|
||||
ray.get(expensive_task.remote())
|
||||
end = time.time()
|
||||
assert end - start < 3
|
||||
|
||||
oids = []
|
||||
for _ in range(3):
|
||||
oid = expensive_task.remote()
|
||||
|
||||
Reference in New Issue
Block a user