File tree Expand file tree Collapse file tree 1 file changed +1
-3
lines changed
verl/workers/rollout/trtllm_rollout Expand file tree Collapse file tree 1 file changed +1
-3
lines changed Original file line number Diff line number Diff line change 3131from torch .distributed .device_mesh import DeviceMesh , init_device_mesh
3232from torch .multiprocessing .reductions import reduce_tensor
3333
34- from verl .utils .memory_utils import aggressive_empty_cache
3534from verl .utils .net_utils import is_valid_ipv6_address
3635from verl .workers .config import HFModelConfig , RolloutConfig
3736from verl .workers .rollout .base import BaseRollout
@@ -424,5 +423,4 @@ async def flush():
424423 if self .is_leader_rank :
425424 # Finalize update weights
426425 await self ._adapter .update_weights (None )
427- await asyncio .to_thread (dist .barrier , group = self .hybrid_device_mesh ["exclude_dp" ].get_group ())
428- aggressive_empty_cache (force_sync = False )
426+ await asyncio .to_thread (dist .barrier , group = self .hybrid_device_mesh ["exclude_dp" ].get_group ())
You can’t perform that action at this time.
0 commit comments