tree 9f7948792f35f7b02c12b0202624b5da12c5c2b2
parent 8d81e61fb644f3fe5ca2993b9cca385fe570baa0
author ydwu4 <ydwu2014@gmail.com> 1707940588 -0800
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> 1708136315 +0000

[export] manuually create test cases for split and inline (#119914)

This PR makes the tests for inline and sequential_split stop relying on set_grad_enabled to be in the graph. Because they'll be gone if we turn on the replace_set_grad_with_hop_pass in the following diff. Instead, we'll manually insert them into the graph.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/119914
Approved by: https://github.com/tugsbayasgalan
ghstack dependencies: #119732, #119736, #119810, #119913