[resubmit] masked_scatter thrust->cub (#58865)
Summary:
See ae7760cf50bb2cddff4663a07b9d68decf4b6c75 for the fix
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58865
Reviewed By: mruberry
Differential Revision: D28657940
Pulled By: ngimel
fbshipit-source-id: 9155c710b0e18ebb3bfa2dabfdd117355ac30840
