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

Update batch_norm_backward_elemt args in native_functions.yaml (#104529)

Browse Source 
cuda kernel impl alread change this name
Pull Request resolved: https://github.com/pytorch/pytorch/pull/104529
Approved by: https://github.com/soulitzer
This commit is contained in:
Dunyang Chen 2023-07-17 21:25:47 +00:00 committed by PyTorch MergeBot
parent 37f5d7866c
commit 0af287cef2
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
aten/src/ATen/native/native_functions.yaml
View File

@ -4157,7 +4157,7 @@
CUDA: batch_norm_backward_reduce_cuda
autogen: batch_norm_backward_reduce.out
- func: batch_norm_backward_elemt(Tensor grad_out, Tensor input, Tensor mean, Tensor invstd, Tensor? weight, Tensor mean_dy, Tensor mean_dy_xmu, Tensor count) -> Tensor
- func: batch_norm_backward_elemt(Tensor grad_out, Tensor input, Tensor mean, Tensor invstd, Tensor? weight, Tensor sum_dy, Tensor sum_dy_xmu, Tensor count) -> Tensor
dispatch:
CUDA: batch_norm_backward_elemt_cuda
autogen: batch_norm_backward_elemt.out

3
test/forward_backward_compatibility/check_forward_backward_compatibility.py
View File

@ -312,7 +312,8 @@ ALLOW_LIST = [
("aten::to_sparse_csc.out", datetime.date(2023, 12, 31)),
("aten::to_sparse_csr.out", datetime.date(2023, 12, 31)),
("aten::_structured_sparse_linear", datetime.date(2023, 12, 31)),
("aten::batch_norm_backward_elemt.out", datetime.date(2023, 12, 31)),
("aten::batch_norm_backward_elemt", datetime.date(2023, 12, 31)),
]
ALLOW_LIST_COMPILED = [