[backend_api] Delete partitioner tags after lowering (#9501)
There was a slight bug i found when lowering with: XnnpackDynamicallyQuantizedPartitioner --> XnnpackPartitioner. The issue arrises because the second time we partition, delegation tags from the previous partitioner still exist. Specifically they exist on the getitem node because the metadata was propagated. See: https://github.com/pytorch/pytorch/blob/main/torch/fx/passes/utils/fuser_utils.py#L235 Since the getitem nodes created in by the XnnpackDynamicallyQuantizedPartitioner still have the delegation tags, they sneak into partitions from XnnpackPartitioner. This shouldn't happen because delegation tags from previous Partitioner should not exist after the previous lowering. We update the code in to_backend here to erase all the "delegation_tags" from node.meta after we've performed all _partition_and_lower Differential Revision: D71518362
P
pytorchbot committed
a0bc7eaf0690a7243ce2f3c9b01bc89f0998d9db
Parent: 922a508
Committed by GitHub <noreply@github.com>
on 3/21/2025, 6:46:13 PM