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

Commit
2 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: bdhirsh Differential Revision: D33217916 Pulled By: albanD fbshipit-source-id: 90f1868f72644f1b5aa023cefd6a102bbbec95af
Author
Parents
Loading