pytorch

707cde9b - [DTensor][Test] Improve math_ops test (#118956)

[DTensor][Test] Improve math_ops test (#118956) The DTensor fully_shard_tensor was created but not used in shard_math_ops test previously. Pull Request resolved: https://github.com/pytorch/pytorch/pull/118956 Approved by: https://github.com/wanchaol