frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Put prim ops used in full jit only in a separate file (#35232)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/35232

Some prim operators, like profile and fusion, are not used in mobile (at least in short term). They are coupled with JIT code. Put them in a separate file (register_prim_ops_fulljit.cpp).
ghstack-source-id: 100807055

Test Plan: buck build //xplat/caffe2:torch

Reviewed By: dreiss

Differential Revision: D20408827

fbshipit-source-id: 9013093357cf75723ef00c34bbfdb6b7ea40a4cf
This commit is contained in:
Martin Yuan
2020-03-25 14:06:49 -07:00
committed by Facebook GitHub Bot
parent 17abb7c31a
commit a4ea16dbc6
4 changed files with 311 additions and 237 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
caffe2/CMakeLists.txt
View File

@ -441,6 +441,7 @@ if(NOT INTERN_BUILD_MOBILE OR NOT BUILD_CAFFE2_MOBILE)
${TORCH_SRC_DIR}/csrc/jit/runtime/print_handler.cpp
${TORCH_SRC_DIR}/csrc/jit/codegen/fuser/interface.cpp
${TORCH_SRC_DIR}/csrc/jit/runtime/register_prim_ops.cpp
${TORCH_SRC_DIR}/csrc/jit/runtime/register_prim_ops_fulljit.cpp
${TORCH_SRC_DIR}/csrc/jit/runtime/register_prim_ops_c10.cpp
${TORCH_SRC_DIR}/csrc/jit/runtime/register_string_ops.cpp
${TORCH_SRC_DIR}/csrc/jit/runtime/register_special_ops.cpp