Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
39 commits
Select commit Hold shift + click to select a range
c4021e1
Initial implementation
sbak5 Jan 20, 2023
2dad201
fix broken docs build
bryevdv Jan 23, 2023
3b16b86
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Mar 14, 2023
62011ae
test boilerplate
bryevdv Mar 14, 2023
ee5c99c
rely on add_boilerplate for conversion
bryevdv May 16, 2023
e7b5611
add check_writeable more places
bryevdv May 16, 2023
015528d
check_writeabe for _contract methods and einsum
bryevdv May 16, 2023
ef73b9f
update broadcast_shapes availability
bryevdv May 16, 2023
dcf404b
Apply suggestions from code review
bryevdv May 16, 2023
8675d41
Merge branch 'bv/broadcast' of github.com:bryevdv/cunumeric into bv/b…
bryevdv May 16, 2023
36d097d
fix broken assertions
bryevdv May 16, 2023
686bdc1
remove unused function
bryevdv May 16, 2023
1840f21
add missing docstrings
bryevdv May 16, 2023
93f6f33
remove owndata assert
bryevdv May 16, 2023
f70d397
revert add_boilerplate for einsum
bryevdv May 16, 2023
74862aa
implement broadcast_to similar to swapaxes
bryevdv May 16, 2023
62aa1fc
update _zip_indices
bryevdv May 16, 2023
7052eea
Apply suggestions from code review
bryevdv May 16, 2023
7ca60bd
wip flagsobj
bryevdv May 18, 2023
6437469
Merge branch 'branch-23.05' into bv/broadcast
bryevdv May 18, 2023
b24d23e
drastically simplify flags tests
bryevdv May 19, 2023
73ec936
add repr
bryevdv May 19, 2023
2927b68
skip owndata test
bryevdv May 19, 2023
e156918
Merge branch 'branch-23.07' into bv/broadcast
bryevdv May 19, 2023
83b7595
Merge branch 'branch-23.07' into bv/broadcast
bryevdv May 31, 2023
151a1cd
Merge branch 'branch-23.07' into bv/broadcast
bryevdv Jun 12, 2023
4fb6e27
flagsobj tweaks
bryevdv Jun 14, 2023
05c0565
clarify flags are for materialized numpy arrays
bryevdv Jun 14, 2023
31485a9
remove redundant check_writeable
bryevdv Jun 14, 2023
63cc899
einsum conversion tweak
bryevdv Jun 14, 2023
d2ed6ff
move check_writeable to array put()
bryevdv Jun 14, 2023
07b4360
only track writeable
bryevdv Jun 14, 2023
0fd07f5
propagate writeable cn -> np
bryevdv Jun 14, 2023
cf40d26
More flags tests, plus minor fixes
manopapad Jun 15, 2023
38a0893
Fix handling of non-ndarray classes in asarray
manopapad Jun 20, 2023
6ac4be3
Merge branch 'branch-23.07' into bv/broadcast
bryevdv Jun 23, 2023
5385dbc
Merge branch 'branch-23.07' into bv/broadcast
bryevdv Jun 23, 2023
f80bc44
fix flags issues
bryevdv Jun 23, 2023
5d86dae
Merge branch 'branch-23.07' into bv/broadcast
bryevdv Jun 23, 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
4 changes: 3 additions & 1 deletion cunumeric/_ufunc/ufunc.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@

import numpy as np

from ..array import convert_to_cunumeric_ndarray, ndarray
from ..array import check_writeable, convert_to_cunumeric_ndarray, ndarray
from ..config import BinaryOpCode, UnaryOpCode, UnaryRedCode
from ..types import NdShape

Expand Down Expand Up @@ -327,6 +327,7 @@ def _prepare_operands(
f"{out.shape} doesn't match the broadcast shape "
f"{out_shape}"
)
check_writeable(out)

if not isinstance(where, bool) or not where:
raise NotImplementedError(
Expand Down Expand Up @@ -648,6 +649,7 @@ def __call__(
arrs, (out,), out_shape, where = self._prepare_operands(
*args, out=out, where=where
)

orig_args = args[: self.nin]

# If no dtype is given to prescribe the accuracy, we use the dtype
Expand Down
Loading