update rocm.py

vllm/platforms/rocm.py

@@ -225,8 +225,9 @@ class RocmPlatform(Platform):
                                  device: Optional[torch.types.Device] = None
                                  ) -> float:
         torch.cuda.reset_peak_memory_stats(device)
-        return torch.cuda.mem_get_info(device)[1] - torch.cuda.mem_get_info(
-            device)[0]
+        # return torch.cuda.mem_get_info(device)[1] - torch.cuda.mem_get_info(
+        #     device)[0]
+        return torch.cuda.max_memory_allocated(device)
 
     @classmethod
     def get_device_communicator_cls(cls) -> str:
@@ -247,4 +248,4 @@ class RocmPlatform(Platform):
         if cls.is_fp8_fnuz():
             return torch.float8_e4m3fnuz
         else:
-            return torch.float8_e4m3fn
+            return torch.float8_e4m3fn
\ No newline at end of file