Skip to content

Commit 8761c77

Browse files
Replace use of broadcastable with shape in aesara.sparse.rewriting
1 parent 7d72236 commit 8761c77

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

aesara/sparse/rewriting.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -677,7 +677,7 @@ def make_node(self, alpha, x_val, x_ind, x_ptr, x_nrows, y, z):
677677
assert x_ind.dtype == "int32"
678678
assert x_ptr.dtype == "int32"
679679
assert x_nrows.dtype == "int32"
680-
assert alpha.ndim == 2 and alpha.type.broadcastable == (True, True)
680+
assert alpha.ndim == 2 and alpha.type.shape == (1, 1)
681681
assert x_val.ndim == 1
682682
assert y.ndim == 2
683683
assert z.ndim == 2
@@ -905,7 +905,7 @@ def c_code_cache_version(self):
905905
{
906906
"pattern": "alpha",
907907
"constraint": lambda expr: (
908-
all(expr.type.broadcastable) and config.blas__ldflags
908+
all(s == 1 for s in expr.type.shape) and config.blas__ldflags
909909
),
910910
},
911911
(sparse._dot, "x", "y"),

0 commit comments

Comments
 (0)