Add smoke test Using a simple RN50 with torch.compile #7359
Conversation
test/smoke_test.py
Outdated
| out = model(x) | ||
| print(out.shape) |
There was a problem hiding this comment.
pytest will swallow all outputs unless the test is failing. Can we maybe assert the expected shape?
There was a problem hiding this comment.
@pmeier We basically want to see if compile works in this PR. As per: pytorch/builder#1318
Will do more testing of this and perhaps add assert as you suggest in the next PR.
NicolasHug
left a comment
NicolasHug
left a comment
There was a problem hiding this comment.
Stamping to unblock but @pmeier comments seem relevant
|
Hey @atalman! 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 |
|
@atalman should we cherry-pick that into the release branch immediately, or do we need to made for a new RC to be cut first? |
|
@NicolasHug no not necessary for cherry pick since we are running tests from main. |
Reviewed By: vmoens Differential Revision: D44416574 fbshipit-source-id: 56a43636a40751a588aef96f22346304f6f29995
4 participants
Add smoke test based on: pytorch/builder#1318