본문 바로가기

dev_AI_framework

Dense / GEMM 역전파 (bias grad) 스케일 - 축 불일치 수정

문제 원인 : 네이티브 Dense bwd 경로에서 bias grad ( db ) 가 평균, 또는 perM 축으로 누적되는 경우 발생, 

gBias 버퍼를 (1, N) 으로 강제 생성, PerN 축으로 고정nse bwd 경로에서 bias grad ( db ) 가 평균, 또는 perM 축으로 누적되는 경우 발생, 

 

gBias 버퍼를 (1, N) 으로 강제 생성, PerN 축으로 고정, gBias 가 PerM 길이로 들어오면 에러