pytorch
2afed243 - [fx2trt] remove split.py (#71933)

Commit
3 years ago
[fx2trt] remove split.py (#71933) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71933 Add the functionalities provided by split.py to splitter_base. - Propagate submodule inputs - Create SplitResult to hold the split results. Then removed split.py, to me this makes navigating the lowering code a bit easier. Added default split and trace function for use. Next step is to add better error handling for each stage during lowering and create unit tests for each stage. I'll probably make some bootcamp tasks for unit tests. Test Plan: CI Reviewed By: frank-wei, wushirong Differential Revision: D33794322 fbshipit-source-id: f991893047a3701177f54cf22d9a6e48e0529472 (cherry picked from commit 1f3e13efbaf93968194180c987e533c0f28bafbd)
Author
Committer
Parents
Loading