mps fix#1227

Jerry-Master wants to merge 3 commits intometa-pytorch:mastermeta-pytorch/captum:masterfrom

Jerry-Master:masterJerry-Master/captum:masterCopy head branch name to clipboard