Skip to content
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

- Required changes for kernels #3165

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

- Required changes for kernels #3165

wants to merge 1 commit into from

Commits on Sep 24, 2024

  1. - Required changes for kernels (pytorch#3165) 

    Summary:
Pull Request resolved: pytorch#3165

X-link: facebookresearch/FBGEMM#259

Adding small changes to kernels for CompiledAutograd support.

Adding `static constexpr bool is_traceable = true;` on kernels, making some kernels to use tensors instead of double and unrolling input shapes on GroupIndexSelectDim0GPUOp from vector into the ctx dict to help enablement of CompiledAutograd.

Reviewed By: Microve

Differential Revision: D63151913
    @flaviotruzzi
    flaviotruzzi committed Sep 24, 2024
    Configuration menu
    Copy the full SHA
    f636ade View commit details
    Browse the repository at this point in the history