已合并
[master] fix sparse ut #14232
AtomGit-Bot创建于 2024年8月31日
[master] fix sparse ut #14232
已合并
从refs/pull/14232/head合入到master
共 1 个文件变更+1-2
| @@ -40,10 +40,9 @@ class TestSparseCoo(TestCase): | |||
| 40 | sparse_coo_npu_neg = sparse_coo_npu.neg() | 40 | sparse_coo_npu_neg = sparse_coo_npu.neg() |
| 41 | self.assertRtolEqual(sparse_coo_cpu_neg._values(), sparse_coo_npu_neg._values()) | 41 | self.assertRtolEqual(sparse_coo_cpu_neg._values(), sparse_coo_npu_neg._values()) |
| 42 | 42 | ||
| 43 | - | ||
| 44 | def test_sparse_coo_copy_(self): | 43 | def test_sparse_coo_copy_(self): |
| 45 | sparse_coo_npu, _ = self._create_sparse_coo_tensor() | 44 | sparse_coo_npu, _ = self._create_sparse_coo_tensor() |
| 46 | - copy_sparse_tensor = torch.rand(sparse_coo_npu.shape).to_sparse().npu().to(sparse_coo_npu.dtype) | 45 | + copy_sparse_tensor = torch.rand(sparse_coo_npu.shape).to(sparse_coo_npu.dtype).to_sparse().npu() |
| 47 | copy_sparse_tensor.copy_(sparse_coo_npu) | 46 | copy_sparse_tensor.copy_(sparse_coo_npu) |
| 48 | self.assertRtolEqual(copy_sparse_tensor._values(), sparse_coo_npu._values()) | 47 | self.assertRtolEqual(copy_sparse_tensor._values(), sparse_coo_npu._values()) |
| 49 | self.assertRtolEqual(copy_sparse_tensor._indices(), sparse_coo_npu._indices()) | 48 | self.assertRtolEqual(copy_sparse_tensor._indices(), sparse_coo_npu._indices()) |