OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

BE: Swap functorch --> torch._higher_order_ops (#152620)

Browse Source 
Summary: Discovered when attempting to resolve arvr builds, should resolve issues around utilizing functorch through export.

Test Plan:
```
buck2 test arvr/mode/linux/opt //arvr/libraries/xrrp/ml/python/test:convert_to_etvk_test
```

Differential Revision: D74013898

Pull Request resolved: https://github.com/pytorch/pytorch/pull/152620
Approved by: https://github.com/zou3519
This commit is contained in:
Eli Uriegas 2025-05-01 21:53:20 +00:00 committed by PyTorch MergeBot
parent 1c04ea4e59
commit b6c5886d09
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torch/_export/pass_base.py
View File

@ -6,7 +6,7 @@ from contextlib import nullcontext
from typing import Any, Callable, Optional, Union
import torch
from functorch.experimental.control_flow import _unstack_pytree
from torch._higher_order_ops.map import _unstack_pytree
from torch import fx
from torch._dispatch.python import enable_python_dispatcher
from torch._export.pass_infra.node_metadata import NodeMetadata