Fix UT from test_register_sharding-pytorch-AtomGit

Star857

Star857

已合并

Fix UT from test_register_sharding #29157

zhangguoguang创建于 1月6日从zhangguoguang/pytorch_npu： fix_UT_290 合入到 Ascend/pytorch： v2.9.0

Fix UT from test_register_sharding #29157

已合并

zhangguoguang创建于 1月6日

从zhangguoguang/pytorch_npu fix_UT_290合入到Ascend/pytorch v2.9.0