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

Add Python to CompositeImplicitAutograd (#82333)

Browse Source 
Signed-off-by: Edward Z. Yang <ezyangfb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82333
Approved by: https://github.com/zou3519
This commit is contained in:
Edward Z. Yang
2022-07-28 07:10:15 -07:00
committed by PyTorch MergeBot
parent 9943ca3ce6
commit 1a20c69385
3 changed files with 44 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_decomp.py
View File

@ -279,6 +279,8 @@ CROSS_REF_EXCLUDE_SET = {
# CompositeAutogradImplicit
# See https://github.com/pytorch/pytorch/issues/81669
(None, None, "nn.functional.relu6"),
(None, None, "nn.functional.mish"),
(None, None, "nn.functional.silu"),
}