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

Fix https://github.com/pytorch/pytorch/issues/138062 (#138137)

Browse Source 
Fixes https://github.com/pytorch/pytorch/issues/138062

Pull Request resolved: https://github.com/pytorch/pytorch/pull/138137
Approved by: https://github.com/mlazos
This commit is contained in:
William Wen 2024-10-17 07:12:15 +00:00 committed by PyTorch MergeBot
parent f3c3f3a3c3
commit 5b7f4767ff
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/compiler/__init__.py
View File

@ -106,7 +106,7 @@ def allow_in_graph(fn):
torch.compiler.allow_in_graph(my_custom_function)
@torch.compile(...)
def fn(a):
def fn(x):
x = torch.add(x, 1)
x = my_custom_function(x)
x = torch.add(x, 1)