-
Notifications
You must be signed in to change notification settings - Fork 7k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
convert torch.split return to list in RAFT #7597
Conversation
Relevant lint CI job: https://github.com/pytorch/vision/actions/runs/5002317531/jobs/8962119940 |
lol pytorch/pytorch#100655 (comment). and yes it needs to support jit :( |
Ok, let me convert to list then. Previously, JIT seemed to have ignored that the return type is |
This reverts commit e6646db.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
stamping, thanks Philip
Hey @pmeier! You merged this PR, but no labels were added. The list of valid labels is available at https://github.com/pytorch/vision/blob/main/.github/process_commit.py |
Reviewed By: vmoens Differential Revision: D46071410 fbshipit-source-id: 043b40a2baee2bc17da658eafb27a55ab8f50c61
pytorch/pytorch#100655 introduced a fix for the annotation. Meaning, the return type always was a tuple, but it was annotated as list. Thus, we also now need to fix our annotation or CI will fail. For example: https://github.com/pytorch/vision/actions/runs/5001904005/jobs/8961199909
I don't know if this needs to be JIT scriptable. If so, this patch is not sufficient, since
Tuple[Tensor, ...]
is not supported by JIT. We need to actually convert into a list then.