The batching rule of aten::affine_grid_generator is not implemented #162346
Description
🚀 The feature, motivation and pitch
I got a warning suggesting me to post an issue here:
UserWarning: There is a performance drop because we have not yet implemented the ba
tching rule for aten::affine_grid_generator. Please file us an issue on GitHub so that we can prioritize its implementation. (Triggered internally at C:\cb\pytorch_1000000000000\work\aten\src\ATen\functorch\BatchedFallback.cpp:85.)
return torch.affine_grid_generator(theta, size, align_corners)
I am using torch.vmap with affine_grid.
Alternatives
No response
Additional context
No response