-
Yu Cheng authored
- Implemented a new role assignment for `AllocateNode` in `warp_specialized_rewriter.cc`, setting the role to `kConsumer` to ensure proper handling of memory allocation scenarios. - This can avoid bug when using T.reduce(clear=False)
5bd3f942