8000 Update base for Update on "Initial support for sparse complex tensor… · pytorch/pytorch@090716d · GitHub
[go: up one dir, main page]
Skip to content

Commit 090716d

Browse files
aocsaaocsa
committed
Update base for Update on "Initial support for sparse complex tensors constructors for CPU/CUDA"
 Currently, sparse tensors only support real floating point tensors. Complex support is added in this PR for CPU/CUDA. - [x] add complex support (torch.cfloat and torch.cdouble) to torch.sparse_coo_tensor constructors - [x] add complex support to coalesce function - [x] add complex support to to_dense function - [x] add complex support to to_sparse function - [x] add complex support to sparse_add function - [x] add unit tests Note: This PR contains only complex support for torch.sparse_coo_tensor fordward function and the related ops used with this function (coalesce, to_dense, to_sparse, and sparse_add). The following PRs in ghstack should cover other sparse operations to have a more complex sparse support, specifically related with the use of specific APIs for accelerated linear algebra. Note: Before using ghstack the original PR was #50984 Differential Revision: [D27765618](https://our.internmc.facebook.com/intern/diff/D27765618) [ghstack-poisoned]
1 parent 26cfbc4 commit 090716d

File tree

0 file changed

+0
-0
lines changed

    0 file changed

    +0
    -0
    lines changed

    0 commit comments

    Comments
     (0)
    0