torch.linalg.qr: forward AD support#67268
Conversation
CI Flow Status⚛️ CI FlowRuleset - Version:
You can add a comment to the PR and tag @pytorchbot with the following commands: # ciflow rerun, "ciflow/default" will always be added automatically
@pytorchbot ciflow rerun
# ciflow rerun with additional labels "-l <ciflow/label_name>", which is equivalent to adding these labels manually and trigger the rerun
@pytorchbot ciflow rerun -l ciflow/scheduled -l ciflow/slowFor more information, please take a look at the CI Flow Wiki. |
🔗 Helpful links
💊 CI failures summary and remediationsAs of commit 09921ea (more details on the Dr. CI page): 💚 💚 Looks good so far! There are no failures yet. 💚 💚 This comment was automatically generated by Dr. CI (expand for details).Please report bugs/suggestions to the (internal) Dr. CI Users group. |
|
forward AD tests hang in my environment now, investigating... |
…taved/linalg_qr_forward_AD
| auto dB1 = std::get<0>(at::triangular_solve( | ||
| dA1.mT().matmul(Q.conj()), | ||
| R1, | ||
| /*upper=*/true, | ||
| /*transpose=*/true | ||
| )).mT(); |
There was a problem hiding this comment.
I'll update these once the new lovely triangular_solve is in :)
|
@albanD has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
…taved/linalg_qr_forward_AD
…taved/linalg_qr_forward_AD
|
@albanD , this PR is now unblocked. It could get merged once CIs show green. |
…taved/linalg_qr_forward_AD
…taved/linalg_qr_forward_AD
…taved/linalg_qr_forward_AD
|
@albanD has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
|
Thanks for the quick rebase! |
Summary: As per title. Pull Request resolved: pytorch#67268 Reviewed By: ngimel Differential Revision: D31960517 Pulled By: albanD fbshipit-source-id: bfd1028a8d352f550efb420f9ca609c09f4a7484
As per title.