frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-16 23:44:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix build tests - test_vmap.py::TestVmapOperatorsOpInfoCPU, test_ops.py::TestOperatorsCPU

Browse Source
This commit is contained in:
Kamil Rocki
2025-10-07 15:47:32 +00:00
parent 4471f8e7e1
commit 8d9dff5a44
2 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_decomp.py
View File

@ -401,6 +401,8 @@ CROSS_REF_EXCLUDE_SET = {
# CompositeAutogradImplicit
# See https://github.com/pytorch/pytorch/issues/81669
(None, None, "nn.functional.relu6"),
(None, None, "nn.functional.linear_cross_entropy"),
(None, None, "aten.linear_cross_entropy"),
# This decomp runs before autograd.
(None, None, "nn.functional.rrelu"),
(None, None, "meshgrid"),

14
torch/testing/_internal/common_methods_invocations.py
View File

@ -14860,6 +14860,20 @@ op_db: list[OpInfo] = [
supports_forward_ad=False,
supports_fwgrad_bwgrad=False,
decorators=(onlyCPU,),
skips=(
DecorateInfo(unittest.skip("linear_cross_entropy vmap support pending"),
"TestVmapOperatorsOpInfo",
"test_op_has_batch_rule"),
DecorateInfo(unittest.skip("linear_cross_entropy vmap support pending"),
"TestVmapOperatorsOpInfo",
"test_vmap_exhaustive"),
DecorateInfo(unittest.skip("linear_cross_entropy fake tensor support pending"),
"TestFakeTensor",
"test_fake"),
DecorateInfo(unittest.skip("linear_cross_entropy fake tensor support pending"),
"TestFakeTensor",
"test_fake_autocast"),
),
),
OpInfo('nn.functional.normalize',
dtypes=floating_and_complex_types_and(torch.half, torch.bfloat16),