Some missing fixes left from previous PRs#440

jerryzh168 merged 1 commit intopytorch:mainpytorch/ao:mainfrom

jerryzh168:some-fixesjerryzh168/ao:some-fixesCopy head branch name to clipboard