Skip to content

Commit cdc8739

Browse files
mfoerste4dinodeep
andauthored
BUG: make __matmul__ path also check for rhs deferred (#1658)
Adds the missing check for deferred arguments for the `__matmul__` path. --------- Co-authored-by: Deep Patel <68748791+dinodeep@users.noreply.github.com>
1 parent b5888e8 commit cdc8739

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

cupynumeric/_thunk/eager.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1710,6 +1710,13 @@ def _matmul(
17101710
"""
17111711
Helper method to apply binary ufunc operations.
17121712
"""
1713+
from .._array.array import ndarray
1714+
1715+
self.check_eager_args(
1716+
rhs._thunk if isinstance(rhs, ndarray) else rhs,
1717+
out._thunk if isinstance(out, ndarray) else out,
1718+
)
1719+
17131720
if self.deferred is not None:
17141721
from .._array.array import ndarray
17151722
from .._module.linalg_mvp import matmul

0 commit comments

Comments
 (0)