pytensor icon indicating copy to clipboard operation
pytensor copied to clipboard

Remove restrictive broadcastability check in `local_blockwise_alloc`

Open ricardoV94 opened this issue 1 year ago • 0 comments

Description

This is very similar in flavor to #742

I'm a bit weary of the fact broadcastability can be redefined by our rewrites, specially after user graph manipulations, because it can mean some graphs may allow broadcasting depending on whether certain rewrites happen or not, and these should be static!

I think the proper solution will come through #408, so that we can both know something has shape 1 and is or is not allowed to broadcast

Related Issue

  • [ ] Closes #
  • [ ] Related to #

Checklist

Type of change

  • [ ] New feature / enhancement
  • [ ] Bug fix
  • [ ] Documentation
  • [x] Maintenance
  • [ ] Other (please specify):

ricardoV94 avatar May 07 '24 12:05 ricardoV94