pytorch
c90be037 - Extend Graph Export to NNC, extend script to support CPU (#74076)

Commit
2 years ago
Extend Graph Export to NNC, extend script to support CPU (#74076) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/74076 Extends the repro script to cpu and NNC. As in file: Usage: ``` 1. Run your script and pipe into a log file PYTORCH_JIT_LOG_LEVEL=">>tensorexpr_fuser" python3 my_test.py &> log.txt 2. Run log_extract: log_extract.py log.txt --baseline --nnc ``` Test Plan: Imported from OSS Reviewed By: gchanan Differential Revision: D34946883 Pulled By: eellison fbshipit-source-id: 644012dbbca0b490820ef83e761c06b0dd009e52 (cherry picked from commit 5256c8f3ff8545033d1335cc96d34194abda1370)
Author
Elias Ellison
Committer
Parents
Loading