Performance drop because of not yet implemented batching rule for bincount

pytorch / functorch

functorch is JAX-like composable function transforms for PyTorch.

https://pytorch.org/functorch/

BSD 3-Clause "New" or "Revised" License

1.38k stars 102 forks source link

Performance drop because of not yet implemented batching rule for bincount #1109

Open IosiaLectus opened 1 year ago

IosiaLectus commented 1 year ago

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."