[MISC] rename CudaMemoryProfiler to DeviceMemoryProfiler (#8703)

2024-09-23 01:44:09 +08:00
parent 8ca5051b9a
commit ca2b628b3c
3 changed files with 5 additions and 5 deletions
--- a/vllm/utils.py
+++ b/vllm/utils.py
@@ -757,7 +757,7 @@ def is_pin_memory_available() -> bool:
    return True


-class CudaMemoryProfiler:
+class DeviceMemoryProfiler:

    def __init__(self, device: Optional[torch.types.Device] = None):
        self.device = device