--reset

--dir=FWD_D,BWD_DW
--sdt=bf16:bf16

--stag=abx:any,axb:any,abx:abx,axb:axb
--batch=../../inputs/prelu/shapes_ci
