pytorch
9ad940d9 - Codegen: ADInplaceOrViewType only include operators registered (#68692)

Commit
4 years ago
Codegen: ADInplaceOrViewType only include operators registered (#68692) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/68692 ADInplaceOrViewType is a sharded file, so by only including specific operator headers, we ensure that changing one (non-method) operator only needs one shard to be re-compiled. This also ports the generated code over to the `at::_ops` interface, and the code generator itself to using `write_sharded` instead of re-implementing its own version of sharding. Test Plan: Imported from OSS Reviewed By: jbschlosser, malfet Differential Revision: D32596274 Pulled By: albanD fbshipit-source-id: 400cad0237829720f94d60f9db7acd0e918e202e
Author
Parents
Loading