Skip to content

Commit 1f2c589

Browse files
committed
.DO NOT MERGE: Attempt fix MLX dispatch for Any
1 parent 0a9d495 commit 1f2c589

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytensor/link/mlx/dispatch/elemwise.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ def prod_reduce(x):
9393
@mlx_funcify_CAReduce_scalar_op.register(AND)
9494
def mlx_funcify_CAReduce_scalar_AND(scalar_op, axis):
9595
def all_reduce(x):
96-
return x.all(axis=axis)
96+
return mx.all(x, axis=axis)
9797

9898
return all_reduce
9999

0 commit comments

Comments
 (0)