Allow autocast in training ir export #137287
Conversation
yushangdi
requested review from
angelayi,
avikchaudhuri,
tugsbayasgalan,
ydwu4 and
zhxchen17
as code owners
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/137287
Note: Links to docs will display an error until the docs builds have been completed. ✅ You can merge normally! (2 Unrelated Failures)As of commit 0148c1f with merge base 46f158b ( FLAKY - The following job failed but was likely due to flakiness present on trunk:
BROKEN TRUNK - The following job failed but was present on the merge base:👉 Rebase onto the `viable/strict` branch to avoid these failures
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
This pull request was exported from Phabricator. Differential Revision: D63345767 |
Summary: X-link: pytorch/executorch#5835 hardcode "val" field for autocast (similar to set_grad_enabled), to bypass the verifier check. Test Plan: CI ``` buck2 run 'fbcode//mode/dev-nosan' fbcode//caffe2/test:test_export -- -r test_export_with_autocast ``` Reviewed By: tugsbayasgalan Differential Revision: D63345767
yushangdi
force-pushed
the
export-D63345767
branch
from
34a0cd7 to
0148c1f
Compare
|
This pull request was exported from Phabricator. Differential Revision: D63345767 |
|
@pytorchbot merge -f 'Landed internally' (Initiating merge automatically since Phabricator Diff has merged, using force because this PR might not pass merge_rules.json but landed internally) |
Merge startedYour change will be merged immediately since you used the force (-f) flag, bypassing any CI checks (ETA: 1-5 minutes). Please use Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
4 participants
Summary: hardcode "val" field for autocast (similar to set_grad_enabled), to bypass the verifier check.
Test Plan: CI
Differential Revision: D63345767