8000 [WIP] CSR/ CSC Elemwise by ivirshup · Pull Request #465 · pydata/sparse · GitHub
[go: up one dir, main page]
Skip to content

[WIP] CSR/ CSC Elemwise #465

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
wants to merge 11 commits into
base: main
Choose a base branch
from
Draft

[WIP] CSR/ CSC Elemwise #465

Changes from 1 commit
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
9db444f
Basic type promotion and computation for CSC/ CSR broadcasting
ivirshup Apr 21, 2021
e312914
Fixup test
ivirshup Apr 21, 2021
91671c9
Minor cleanup (might have broke something)
ivirshup Apr 21, 2021
7f2d2ed
temp print statement
ivirshup Apr 22, 2021
5ad790e
Fix return type determination
ivirshup Apr 22, 2021
34ce299
Fix fill_value argument handling for CSC and CSR, handle scipy -> spa…
ivirshup Apr 22, 2021
ccffedc
Unbreak elemwise ops with CSR/CSC and COO
ivirshup Apr 22, 2021
30004b9
Skip some tests with densification
ivirshup Apr 22, 2021
c9510af
Resize array
ivirshup Apr 24, 2021
c96cc1a
Use pruning strategy from scipy instead of resizing (it seems a bit f…
ivirshup Apr 24, 2021
9b7f22c
Merge branch 'master' into elemwise
ivirshup Jul 20, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
temp print statement
  • Loading branch information
@ivirshup
ivirshup committed Apr 22, 2021
commit 7f2d2ed6738c6ffaf1b5fdf20185eeca701966f2
4 changes: 4 additions & 0 deletions sparse/_umath.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -411,8 +411,12 @@ def _resolve_result_type(args: "list[ArrayLike]") -> "Type":
from ._compressed import GCXS, CSR, CSC
from ._coo import COO
from ._dok import DOK
from ._sparse_array import SparseArray
from ._compressed.compressed import _Compressed2d

args = [arg for arg in args if isinstance(arg, SparseArray)]
print([type(arg) for arg in args])

if all(isinstance(arg, DOK) for arg in args):
out_type = DOK
elif all(isinstance(arg, CSR) for arg in args):
Expand Down
0