Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Simplify macros for exposing c10 ops to c2 (#17781)
Summary: Pull Request resolved: pytorch/pytorch#17781 The wrapper for calling a c10 operator from caffe2 is now based on a runtime FunctionSchema instead of compile time information. This way, it can be created for any c10 operator schema with just one invocation to a simple macro instead of having to define arguments and more as compile time structures. Furthermore, previously, the wrapper assumed there's an argument present for preallocated outputs, but that was only true for caffe2 operators exported to c10. So the wrapper only worked correctly for calling caffe2->c10->caffe2. Now with the new implementation, it works for any c10 operator. Also, binary size for this should be much smaller. Reviewed By: ezyang Differential Revision: D14375054 fbshipit-source-id: bac7ab8e63929e6e2a148eacac41ed092009aa86
- Loading branch information