Unverified Commit 94643fac authored by Leo Jiang's avatar Leo Jiang Committed by GitHub
Browse files

[bugfix] bugfix for npu free memory (#9640)



* Improve NPU performance

* Improve NPU performance

* Improve NPU performance

* Improve NPU performance

* [bugfix] bugfix for npu free memory

* [bugfix] bugfix for npu free memory

* [bugfix] bugfix for npu free memory

---------
Co-authored-by: default avatar蒋硕 <jiangshuo9@h-partners.com>
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 435f6b7e
...@@ -284,7 +284,7 @@ def free_memory(): ...@@ -284,7 +284,7 @@ def free_memory():
elif torch.backends.mps.is_available(): elif torch.backends.mps.is_available():
torch.mps.empty_cache() torch.mps.empty_cache()
elif is_torch_npu_available(): elif is_torch_npu_available():
torch_npu.empty_cache() torch_npu.npu.empty_cache()
# Adapted from torch-ema https://github.com/fadel/pytorch_ema/blob/master/torch_ema/ema.py#L14 # Adapted from torch-ema https://github.com/fadel/pytorch_ema/blob/master/torch_ema/ema.py#L14
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment