Commit Graph

5 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
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