xla
use native_layer_norm_backward decomp from pytorch core
#3612
Merged

use native_layer_norm_backward decomp from pytorch core #3612

bdhirsh merged 2 commits into master from layer_norm_decomp_fix
bdhirsh
bdhirsh use native_layer_norm_backward decomp from pytorch core
c86571bc
bdhirsh bdhirsh force pushed from 84ca6b21 to c86571bc 3 years ago
bdhirsh reuse the composite forward kernels from core instead
0b9941b6
bdhirsh
bdhirsh bdhirsh requested a review from JackCaoG JackCaoG 3 years ago
bdhirsh
JackCaoG
JackCaoG
JackCaoG approved these changes on 2022-05-26
bdhirsh
bdhirsh bdhirsh merged 08084ea0 into master 3 years ago
bdhirsh bdhirsh deleted the layer_norm_decomp_fix branch 3 years ago
JackCaoG

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone