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

[while_loop] expose it as torch.while_loop (#128562)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/128562
Approved by: https://github.com/zou3519
This commit is contained in:
Yidi Wu
2024-06-12 15:42:10 -07:00
committed by PyTorch MergeBot
parent c486e2ab64
commit afdaa7fc95
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
torch/__init__.py
View File

@ -2358,7 +2358,7 @@ def compile(
from torch import export as export
from torch._higher_order_ops import cond as cond
from torch._higher_order_ops import cond, while_loop
def _register_device_module(device_type, module):