search

pytorch / xla

Enabling PyTorch on XLA Devices (e.g. Google TPU)
https://pytorch.org/xla
Other
2.49k stars 480 forks source link

Add peak_bytes_used to the get_memory_info #8265

Closed JackCaoG closed 1 month ago

JackCaoG commented 1 month ago

@lsy323 can you take a look at this one?