Skip to content

Commit 5f8ddb3

Browse files
committed
Add developer note to rewrite
1 parent ff56b7f commit 5f8ddb3

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

pytensor/tensor/rewriting/blockwise.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -99,6 +99,8 @@ def local_blockwise_alloc(fgraph, node):
9999
BOp(vector, alloc(vector, 10, 5)) -> alloc(BOp)(vector, vector), 10, 5)
100100
BOp(vector, alloc(scalar, 10, 5)) -> alloc(BOp)(vector, alloc(scalar, 5), 10, 5)
101101
BOp(matrix, alloc(vector, 10, 5)) -> BOp(matrix, vector)
102+
103+
This is critical to remove many unnecessary Blockwise, or to reduce the work done by it
102104
"""
103105

104106
op: Blockwise = node.op # type: ignore

0 commit comments

Comments
 (0)