Skip to content

Commit

Permalink
Replace use of broadcastable with shape in aesara.sparse.rewriting
Browse files Browse the repository at this point in the history
  • Loading branch information
brandonwillard committed Nov 4, 2022
1 parent 6c0fc17 commit b586063
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions aesara/sparse/rewriting.py
Original file line number Diff line number Diff line change
Expand Up @@ -677,7 +677,7 @@ def make_node(self, alpha, x_val, x_ind, x_ptr, x_nrows, y, z):
assert x_ind.dtype == "int32"
assert x_ptr.dtype == "int32"
assert x_nrows.dtype == "int32"
assert alpha.ndim == 2 and alpha.type.broadcastable == (True, True)
assert alpha.ndim == 2 and alpha.type.shape == (1, 1)
assert x_val.ndim == 1
assert y.ndim == 2
assert z.ndim == 2
Expand Down Expand Up @@ -905,7 +905,7 @@ def c_code_cache_version(self):
{
"pattern": "alpha",
"constraint": lambda expr: (
all(expr.type.broadcastable) and config.blas__ldflags
all(s == 1 for s in expr.type.shape) and config.blas__ldflags
),
},
(sparse._dot, "x", "y"),
Expand Down

0 comments on commit b586063

Please sign in to comment.