New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Fix output size computation for MaxPool2D for ceil_model = true. #3890

Open

sahas3 wants to merge 1 commit into llvm:main from sahas3:poolCeil

+69 −5

Contributor

sahas3 commented Nov 22, 2024

This PR fixes the output size computation as per https://github.com/pytorch/pytorch/blob/d8c14838f164ee02b88b6e37471b71bb0373f865/torch/_meta_registrations.py#L3847

if ceil_mode:
        if (outputSize - 1) * stride >= inputSize + pad_l:
            outputSize -= 1
    return outputSize


          Fix output size computation for MaxPool2D for ceil_model = true.

b1550a2

sahas3 requested review from sjarus and vivekkhandelwal1

November 22, 2024 20:02

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet