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

Bump black version to 23.1.0 (#96578)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/96578
Approved by: https://github.com/ezyang
This commit is contained in:
BowenBao
2023-03-14 19:46:45 -07:00
committed by PyTorch MergeBot
parent a229e78544
commit 60a68477a6
114 changed files with 111 additions and 167 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
torch/_lobpcg.py
View File

@ -273,7 +273,6 @@ class LOBPCGAutogradFunction(torch.autograd.Function):
ortho_fparams: Optional[Dict[str, float]] = None,
ortho_bparams: Optional[Dict[str, bool]] = None,
) -> Tuple[Tensor, Tensor]:
# makes sure that input is contiguous for efficiency.
# Note: autograd does not support dense gradients for sparse input yet.
A = A.contiguous() if (not A.is_sparse) else A
@ -360,7 +359,6 @@ def lobpcg(
ortho_fparams: Optional[Dict[str, float]] = None,
ortho_bparams: Optional[Dict[str, bool]] = None,
) -> Tuple[Tensor, Tensor]:
"""Find the k largest (or smallest) eigenvalues and the corresponding
eigenvectors of a symmetric positive definite generalized
eigenvalue problem using matrix-free LOBPCG methods.
@ -598,7 +596,6 @@ def _lobpcg(
ortho_fparams: Optional[Dict[str, float]] = None,
ortho_bparams: Optional[Dict[str, bool]] = None,
) -> Tuple[Tensor, Tensor]:
# A must be square:
assert A.shape[-2] == A.shape[-1], A.shape
if B is not None:
@ -707,7 +704,6 @@ class LOBPCG:
method: str,
tracker: None,
) -> None:
# constant parameters
self.A = A
self.B = B
@ -833,7 +829,6 @@ class LOBPCG:
self.call_tracker()
while not self.stop_iteration():
self.update()
if not torch.jit.is_scripting() and self.tracker is not None: