Commit Graph

6 Commits

Author SHA1 Message Date
Justin Chu
abc1cadddb [BE] Enable ruff's UP rules and autoformat utils/ (#105424)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/105424
Approved by: https://github.com/ezyang, https://github.com/malfet
2023-07-18 20:17:25 +00:00
Nikita Shulga
40e2aadf47 Create __init__.py (#78629)
To make `torch.utils.jit` a proper package, otherwise it will not be added to the wheel

Pull Request resolved: https://github.com/pytorch/pytorch/pull/78629
Approved by: https://github.com/seemethere, https://github.com/xuzhao9, https://github.com/davidberard98
2022-06-03 18:14:21 +00:00
David Berard
fa44e165ff Retry "[JIT] parse prim::Constant[value=annotate()] and prim::Constant[value={0}]"
Retry of https://github.com/pytorch/pytorch/pull/76875. It was reverted
due to torchvision failures, but it turned out that the failures were
caused by a different PR.

irparser previously didn't support these, which would cause failures in
log_extract.py

Pull Request resolved: https://github.com/pytorch/pytorch/pull/77377

Approved by: https://github.com/datumbox
2022-05-13 15:12:07 +00:00
PyTorch MergeBot
2083b16f68 Revert "[JIT] parse prim::Constant[value=annotate()] and prim::Constant[value={0}]"
This reverts commit 31d3ce7000.

Reverted https://github.com/pytorch/pytorch/pull/76875 on behalf of https://github.com/janeyx99
2022-05-11 13:50:20 +00:00
David Berard
31d3ce7000 [JIT] parse prim::Constant[value=annotate()] and prim::Constant[value={0}]
irparser previously didn't support these, which would cause failures in
log_extract.py

Pull Request resolved: https://github.com/pytorch/pytorch/pull/76875

Approved by: https://github.com/eellison, https://github.com/tugsbayasgalan
2022-05-06 19:40:09 +00:00
David Berard
9d05ce602e [JIT] Move log_extract.py helper functions to torch.utils
This will allow us to reuse the log_extract.py tools in torchbench

Pull Request resolved: https://github.com/pytorch/pytorch/pull/75436

Approved by: https://github.com/eellison
2022-04-07 22:17:58 +00:00