Mark attri…
Summary:
Shuai wants to test this internally before https://github.com/pytorch/pytorch/pull/133713 can go in. Creating a separate PR for ghmport.
cc rec
X-link: https://github.com/pytorch/pytorch/pull/134136
Reviewed By: yanboliang
Differential Revision: D61612768
Pulled By: anijain2305
fbshipit-source-id: 9681ee2215967730446b3abe57fbf0f5ed209968
