frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 13:44:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "[BE]: Enable RUFF TRY400 rule - log.exception (#153473)"

Browse Source 
This reverts commit 4f4ecc583e0f48ad2d062a53bf91c61ab40b4948.

Reverted https://github.com/pytorch/pytorch/pull/153473 on behalf of https://github.com/jeanschmidt due to seems to have broken internal signals, @albanD may I count on you to help the author merge his PR? D74837988 ([comment](https://github.com/pytorch/pytorch/pull/153473#issuecomment-2886017075))
This commit is contained in:
PyTorch MergeBot
2025-05-16 08:29:26 +00:00
parent 86c6f71ddb
commit 3443627e07
23 changed files with 46 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/distributed/algorithms/ddp_comm_hooks/powerSGD_hook.py
View File

@ -60,7 +60,7 @@ def _orthogonalize_gram_schmidt(matrices, epsilon=0):
try:
col /= torch.norm(col, dim=1, keepdim=True)
except ZeroDivisionError:
logger.exception(
logger.error(
"The matrices to be orthogonalized has at least a column of all 0s. Please set a small value such as 1e-8 "
"as `orthogonalization_epsilon` in PowerSGD state."
)