`bmm`: improve error handling and error messages.#9643

ysiraichi wants to merge 1 commit intomasterpytorch/xla:masterfrom

ysiraichi/better-error-bmmpytorch/xla:ysiraichi/better-error-bmmCopy head branch name to clipboard