pytorch
6ccb74b8 - Update codegen to use boxed kernel (#63459)

Commit
3 years ago
Update codegen to use boxed kernel (#63459) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/63459 - Replaces the usual registration basically when "requires_derivative" is True (as in we still need a grad_fn), but `fn.info` is `None` (TODO maybe make sure differentiable inputs > 0 also to match requires_derivative). - Adds some (temporary?) fixes to some sparse functions See: https://github.com/pytorch/pytorch/issues/63549 - To remove the codegen that generates NotImplemented node (though that should only be one line), because there are some ops listed under `RESET_GRAD_ACCUMULATOR` that have a extra function call. We would need to make this list of ops available to c++, but this would either mean we'd have to codegen a list of strings, or move the RESET_GRAD_ACCUMULATOR to cpp land. We could do this in a future PR if necessary. Test Plan: Imported from OSS Reviewed By: heitorschueroff Differential Revision: D30518571 Pulled By: soulitzer fbshipit-source-id: 99a35cbced46292d1b4e51594ae4d534c2caf8b6
Author
Parents
Loading