MIOpen icon indicating copy to clipboard operation
MIOpen copied to clipboard

[gTests] Create new gTests for bn_fwd_train and bn_bwd

Open xinlipn opened this issue 9 months ago • 2 comments

Create gTests to cover cases without saved Mean and saved InvVar, equivalent to e.g. the command below

./bin/MIOpenDriver bnormbfp16 -n 1 -c 2 -H 24 -W 16 -m 1 --forw 1 -b 0 -s 0 -r 0 -t 1 --layout NCHW

xinlipn avatar Mar 05 '25 19:03 xinlipn

I do not see bn bwd cpu handle both "saved mean" and "no saved mean"

bghimireamd avatar Mar 05 '25 19:03 bghimireamd

I do not see bn bwd cpu handle both "saved mean" and "no saved mean"

Hi @bghimireamd , that's a good catch. I have this change locally, but but bn_bwd doesn't seem to pass yet with it. I have just committed the change to ComputeCPUBNBwd

xinlipn avatar Mar 06 '25 06:03 xinlipn

Imported to ROCm/rocm-libraries

amd-hsivasun avatar Jul 28 '25 20:07 amd-hsivasun