This repository was archived by the owner on Aug 21, 2025. It is now read-only.
This repository was archived by the owner on Aug 21, 2025. It is now read-only.
Performance drop because of not yet implemented batching rule for bincount #1109
Description
I am currently using functorch.vmap with torch.bincount, which gives the warning
"UserWarning: There is a performance drop because we have not yet implemented the batching rule for aten::bincount. Please file us an issue on GitHub so that we can prioritize its implementation."