frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor partitioner and clean it up (#126318)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/126318
Approved by: https://github.com/anijain2305
This commit is contained in:
chilli
2024-05-16 14:44:55 -07:00
committed by PyTorch MergeBot
parent 5756b53dd8
commit f9a7033194
4 changed files with 668 additions and 605 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
functorch/compile/__init__.py
View File

@ -25,7 +25,6 @@ from torch._functorch.fx_minifier import minifier
from torch._functorch.partitioners import (
default_partition,
draw_graph,
draw_joint_graph,
min_cut_rematerialization_partition,
)
from torch._functorch.python_key import pythonkey_decompose