[BE] [moe training] bench script for single device moe layer#3126

danielvegamyhre merged 1 commit intomainpytorch/ao:mainfrom

single-moepytorch/ao:single-moeCopy head branch name to clipboard