Edward Z. Yang
|
d4f065d261
|
Return mode object from __enter__ (#80998)
This makes `with Mode() as m:` work.
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/80998
Approved by: https://github.com/samdow
|
2022-07-12 23:22:26 +00:00 |
|
Edward Z. Yang
|
3ca309c4b8
|
Correctly setup ancestors on explicit push mode. (#80995)
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/80995
Approved by: https://github.com/Chillee
|
2022-07-07 02:01:12 +00:00 |
|
samdow
|
5e926aafab
|
add utils for checking that all modes are in the same scope and finding the outermost mode
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78847
Approved by: https://github.com/ezyang, https://github.com/zou3519
|
2022-06-10 19:31:05 +00:00 |
|
samdow
|
3734fcc8f8
|
add ability to push a mode if the current mode is an ancestor
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78822
Approved by: https://github.com/ezyang, https://github.com/zou3519
|
2022-06-10 18:27:04 +00:00 |
|
samdow
|
aa06d05297
|
enable with semantics
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78214
Approved by: https://github.com/ezyang, https://github.com/zou3519
|
2022-06-01 21:14:45 +00:00 |
|
Elias Ellison
|
678213ead2
|
Fake Tensor Part 1
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77969
Approved by: https://github.com/ezyang
|
2022-05-31 16:20:35 +00:00 |
|
samdow
|
6779366f27
|
add nested mode to python mode
Pull Request resolved: https://github.com/pytorch/pytorch/pull/75965
Approved by: https://github.com/albanD, https://github.com/ezyang, https://github.com/zou3519
|
2022-05-04 13:01:06 +00:00 |
|