Skip to content

Fix layer_norm backward to reuse the forward dropout mask and scale dx1 by rowscale - #1023

Open
cyboTiger wants to merge 1 commit into
state-spaces:mainfrom
cyboTiger:main
Open

Fix layer_norm backward to reuse the forward dropout mask and scale dx1 by rowscale#1023
cyboTiger wants to merge 1 commit into
state-spaces:mainfrom
cyboTiger:main

fix[triton]: fix layer_norm backward to reuse the forward dropout mas…

bcd54b6
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs