[c2] cuda impl for WeightScale op (#38712)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/38712
as title
Test Plan: buck test;
Reviewed By: ustctf
Differential Revision: D21586705
fbshipit-source-id: 12cd34f04f074ee12b77304055f3ba6068cf38fb