Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
ZelboK committed May 14, 2024
1 parent fec9793 commit 489afbe
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion aten/src/ATen/native/BatchLinearAlgebra.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -3437,7 +3437,7 @@ static void linalg_lstsq_out_info(
} else {
auto [U, S, Vh] = at::_linalg_svd(input, false, true, "gesvd");
rank = at::zeros({1}, at::kLong);

auto S_pinv = S.reciprocal();
auto s1 = at::narrow(S, /*dim=*/-1, /*start=*/0, /*length=*/1); // singular values are sorted in descending order
S_pinv.masked_fill_(S < rcond * s1, 0);
Expand Down

0 comments on commit 489afbe

Please sign in to comment.