minor

2025-11-04 08:00:58 +08:00 · 2025-09-10 10:06:54 +00:00
1 changed files with 1 additions and 0 deletions
--- a/test/test_sparse.py
+++ b/test/test_sparse.py
@ -4019,6 +4019,7 @@ class TestSparse(TestSparseBase):

            # Case 1: sparse broadcasts over dense
            s = self._gen_sparse(sparse_dim, nnz, sub_shape, dtype, device, coalesced)[0]
+            s = torch._sparse_broadcast_to(s, shape)
            d = make_tensor(shape, dtype=dtype, device=device)
            check(self, s, d)
            check_empty(sub_shape, nnz, shape, coalesced)