Skip to content

Commit

Permalink
fixed issue on amdgpus
Browse files Browse the repository at this point in the history
  • Loading branch information
parthraut committed Apr 30, 2024
1 parent a64d9c5 commit 7f611c8
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions zeus/device/gpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -815,8 +815,8 @@ def __init__(self, ensure_homogeneous: bool = False) -> None:
raise exception_class(e.msg) from e

@property
def gpus(self) -> list[GPU]:
"""Returns a list of NVIDIAGPU objects being tracked."""
def gpus(self) -> Sequence[GPU]:
"""Returns a list of AMDGPU objects being tracked."""
return self._gpus

def _init_gpus(self) -> None:
Expand All @@ -834,10 +834,6 @@ def __del__(self) -> None:
"""Shuts down the AMD GPU monitoring library to release resources and clean up."""
with contextlib.suppress(amdsmi.AmdSmiException):
amdsmi.amdsmi_shut_down() # Ignore error on shutdown. Neccessary for proper cleanup and test functionality
@property
def gpus(self) -> Sequence[GPU]:
"""Returns a list of AMDGPU objects being tracked."""
raise NotImplementedError("AMDGPUs.gpus is not implemented yet.")


_gpus: GPUs | None = None
Expand Down

0 comments on commit 7f611c8

Please sign in to comment.