Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Addsemiring(A @ B @ C) that applies semiring to both matmuls#501

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.

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

Already on GitHub?Sign in to your account

Merged
eriknw merged 18 commits intopython-graphblas:mainfromeriknw:multi_arg_semiring
Nov 4, 2023
Merged
Show file tree
Hide file tree
Changes fromall commits
Commits
Show all changes
18 commits
Select commitHold shift + click to select a range
6e8814b
Add `semiring(A @ B @ C)` that applies semiring to both matmuls
eriknwSep 5, 2023
7d73cb6
WIP: begin implementing ewise add too
eriknwSep 8, 2023
6285d83
Fix imports
eriknwSep 8, 2023
d8a5f47
Merge branch 'main' into multi_arg_semiring
eriknwSep 12, 2023
fa61888
checkpoint
eriknwSep 14, 2023
9b1fdb1
a little more
eriknwSep 14, 2023
8a6333a
Merge branch 'main' into multi_arg_semiring
eriknwSep 22, 2023
84ae93e
Merge branch 'main' into multi_arg_semiring
eriknwSep 29, 2023
ef74652
Merge branch 'main' into multi_arg_semiring
eriknwSep 29, 2023
0a26c3e
bump pre-commit
eriknwSep 29, 2023
2623bef
Merge branch 'multi_arg_semiring2' into multi_arg_semiring
eriknwSep 29, 2023
314c190
Merge branch 'main' into multi_arg_semiring
eriknwOct 15, 2023
5396330
More tests
eriknwOct 15, 2023
b278d2b
Merge branch 'main' into multi_arg_semiring
eriknwOct 28, 2023
343081e
more tests
eriknwOct 28, 2023
ecaf6ac
Getting very close I think
eriknwOct 29, 2023
18a1e1b
Clean up, and handle dimension mismatch more explicitly in infix exprs
eriknwOct 30, 2023
baf4709
Merge branch 'main' into multi_arg_semiring
eriknwOct 30, 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
16 changes: 12 additions & 4 deletionsgraphblas/core/base.py
View file
Open in desktop
Original file line numberDiff line numberDiff line change
Expand Up@@ -263,23 +263,31 @@ def __call__(
)

def __or__(self, other):
from .infix import _ewise_infix_expr
from .infix import _ewise_infix_expr, _ewise_mult_expr_types

if isinstance(other, _ewise_mult_expr_types):
raise TypeError("XXX")
return _ewise_infix_expr(self, other, method="ewise_add", within="__or__")

def __ror__(self, other):
from .infix import _ewise_infix_expr
from .infix import _ewise_infix_expr, _ewise_mult_expr_types

if isinstance(other, _ewise_mult_expr_types):
raise TypeError("XXX")
return _ewise_infix_expr(other, self, method="ewise_add", within="__ror__")

def __and__(self, other):
from .infix import _ewise_infix_expr
from .infix import_ewise_add_expr_types,_ewise_infix_expr

if isinstance(other, _ewise_add_expr_types):
raise TypeError("XXX")
return _ewise_infix_expr(self, other, method="ewise_mult", within="__and__")

def __rand__(self, other):
from .infix import _ewise_infix_expr
from .infix import_ewise_add_expr_types,_ewise_infix_expr

if isinstance(other, _ewise_add_expr_types):
raise TypeError("XXX")
return _ewise_infix_expr(other, self, method="ewise_mult", within="__rand__")

def __matmul__(self, other):
Expand Down
72 changes: 72 additions & 0 deletionsgraphblas/core/infix.py
View file
Open in desktop
Original file line numberDiff line numberDiff line change
Expand Up@@ -126,6 +126,19 @@ class ScalarEwiseAddExpr(ScalarInfixExpr):

_to_expr = _ewise_add_to_expr

# Allow e.g. `plus(x | y | z)`
__or__ = Scalar.__or__
__ror__ = Scalar.__ror__
_ewise_add = Scalar._ewise_add
_ewise_union = Scalar._ewise_union

# Don't allow e.g. `plus(x | y & z)`
def __and__(self, other):
raise TypeError("XXX")

def __rand__(self, other):
raise TypeError("XXX")


class ScalarEwiseMultExpr(ScalarInfixExpr):
__slots__ = ()
Expand All@@ -135,6 +148,18 @@ class ScalarEwiseMultExpr(ScalarInfixExpr):

_to_expr = _ewise_mult_to_expr

# Allow e.g. `plus(x & y & z)`
__and__ = Scalar.__and__
__rand__ = Scalar.__rand__
_ewise_mult = Scalar._ewise_mult

# Don't allow e.g. `plus(x | y & z)`
def __or__(self, other):
raise TypeError("XXX")

def __ror__(self, other):
raise TypeError("XXX")


class ScalarMatMulExpr(ScalarInfixExpr):
__slots__ = ()
Expand DownExpand Up@@ -239,6 +264,15 @@ class VectorEwiseAddExpr(VectorInfixExpr):

_to_expr = _ewise_add_to_expr

# Allow e.g. `plus(x | y | z)`
__or__ = Vector.__or__
__ror__ = Vector.__ror__
_ewise_add = Vector._ewise_add
_ewise_union = Vector._ewise_union
# Don't allow e.g. `plus(x | y & z)`
__and__ = ScalarEwiseAddExpr.__and__ # raises
__rand__ = ScalarEwiseAddExpr.__rand__ # raises


class VectorEwiseMultExpr(VectorInfixExpr):
__slots__ = ()
Expand All@@ -248,6 +282,14 @@ class VectorEwiseMultExpr(VectorInfixExpr):

_to_expr = _ewise_mult_to_expr

# Allow e.g. `plus(x & y & z)`
__and__ = Vector.__and__
__rand__ = Vector.__rand__
_ewise_mult = Vector._ewise_mult
# Don't allow e.g. `plus(x | y & z)`
__or__ = ScalarEwiseMultExpr.__or__ # raises
__ror__ = ScalarEwiseMultExpr.__ror__ # raises


class VectorMatMulExpr(VectorInfixExpr):
__slots__ = "method_name"
Expand All@@ -259,6 +301,11 @@ def __init__(self, left, right, *, method_name, size):
self.method_name = method_name
self._size = size

__matmul__ = Vector.__matmul__
__rmatmul__ = Vector.__rmatmul__
_inner = Vector._inner
_vxm = Vector._vxm


utils._output_types[VectorEwiseAddExpr] = Vector
utils._output_types[VectorEwiseMultExpr] = Vector
Expand DownExpand Up@@ -376,6 +423,15 @@ class MatrixEwiseAddExpr(MatrixInfixExpr):

_to_expr = _ewise_add_to_expr

# Allow e.g. `plus(x | y | z)`
__or__ = Matrix.__or__
__ror__ = Matrix.__ror__
_ewise_add = Matrix._ewise_add
_ewise_union = Matrix._ewise_union
# Don't allow e.g. `plus(x | y & z)`
__and__ = VectorEwiseAddExpr.__and__ # raises
__rand__ = VectorEwiseAddExpr.__rand__ # raises


class MatrixEwiseMultExpr(MatrixInfixExpr):
__slots__ = ()
Expand All@@ -385,6 +441,14 @@ class MatrixEwiseMultExpr(MatrixInfixExpr):

_to_expr = _ewise_mult_to_expr

# Allow e.g. `plus(x & y & z)`
__and__ = Matrix.__and__
__rand__ = Matrix.__rand__
_ewise_mult = Matrix._ewise_mult
# Don't allow e.g. `plus(x | y & z)`
__or__ = VectorEwiseMultExpr.__or__ # raises
__ror__ = VectorEwiseMultExpr.__ror__ # raises


class MatrixMatMulExpr(MatrixInfixExpr):
__slots__ = ()
Expand All@@ -397,6 +461,11 @@ def __init__(self, left, right, *, nrows, ncols):
self._nrows = nrows
self._ncols = ncols

__matmul__ = Matrix.__matmul__
__rmatmul__ = Matrix.__rmatmul__
_mxm = Matrix._mxm
_mxv = Matrix._mxv


utils._output_types[MatrixEwiseAddExpr] = Matrix
utils._output_types[MatrixEwiseMultExpr] = Matrix
Expand DownExpand Up@@ -514,5 +583,8 @@ def _matmul_infix_expr(left, right, *, within):
)


_ewise_add_expr_types = (MatrixEwiseAddExpr, VectorEwiseAddExpr, ScalarEwiseAddExpr)
_ewise_mult_expr_types = (MatrixEwiseMultExpr, VectorEwiseMultExpr, ScalarEwiseMultExpr)

# Import infixmethods, which has side effects
from . import infixmethods # noqa: E402, F401 isort:skip
Loading
[8]ページ先頭
©2009-2025 Movatter.jp