Skip to content

Commit 3e0ab2e

Browse files
author
Allen Wang
committed
remove legacy gpu manager
1 parent fa456c7 commit 3e0ab2e

File tree

4 files changed

+1
-355
lines changed

4 files changed

+1
-355
lines changed

src/forge/controller/provisioner.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -110,12 +110,7 @@ async def set_environment(proc_mesh: ProcMesh, env_vars: dict[str, str]):
110110

111111

112112
class GpuManager:
113-
"""Tracks and assigns GPU devices on a host.
114-
115-
This currently mimics the `gpu_manager` in system_controllers - we will
116-
consolidate as part of the "proper HostMesh integration" work.
117-
118-
"""
113+
"""Tracks and assigns GPU devices on a host."""
119114

120115
def __init__(self, available_devices: set[int] | None = None):
121116
if available_devices is None:

src/forge/controller/system_controllers/__init__.py

Lines changed: 0 additions & 12 deletions
This file was deleted.

src/forge/controller/system_controllers/gpu_manager.py

Lines changed: 0 additions & 73 deletions
This file was deleted.

tests/unit_tests/test_gpu_manager.py

Lines changed: 0 additions & 264 deletions
This file was deleted.

0 commit comments

Comments
 (0)