frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Inductor] support mixed dtype in the native_layer_norm_backward meta function (#159830)

Browse Source 
Fixes #159829

Pull Request resolved: https://github.com/pytorch/pytorch/pull/159830
Approved by: https://github.com/albanD
This commit is contained in:
Shaobin Ma
2025-09-17 20:29:08 +00:00
committed by PyTorch MergeBot
parent dfda2dfd53
commit 63276edb7c
2 changed files with 39 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
torch/_decomp/decompositions.py
View File

@ -1710,8 +1710,8 @@ def native_layer_norm_backward(
return (
_maybe_cast(d_input, input.dtype),
_maybe_cast(d_weight, input.dtype),
_maybe_cast(d_bias, input.dtype),
_maybe_cast(d_weight, weight.dtype if weight is not None else None),
_maybe_cast(d_bias, bias.dtype if bias is not None else None),
)