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

Add and test training in lite interpreter. (#32359)

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

Test Plan: Imported from OSS

Differential Revision: D19450614

Pulled By: iseeyuan

fbshipit-source-id: 6bafff39d7880a5b7fb9cd70c33a4e584812be12
This commit is contained in:
Martin Yuan
2020-03-03 23:31:03 -08:00
committed by Facebook Github Bot
parent 2ba74b741e
commit f097ca503d
9 changed files with 127 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
caffe2/CMakeLists.txt
View File

@ -484,12 +484,13 @@ if (NOT INTERN_BUILD_MOBILE OR NOT BUILD_CAFFE2_MOBILE)
${TORCH_SRC_DIR}/csrc/jit/mobile/type_parser.cpp
)
if (NOT INTERN_DISABLE_MOBILE_INTERP)
if (NOT INTERN_DISABLE_MOBILE_INTERP AND BUILD_CAFFE2_MOBILE)
set (MOBILE_SRCS
${TORCH_SRC_DIR}/csrc/jit/mobile/function.cpp
${TORCH_SRC_DIR}/csrc/jit/mobile/import.cpp
${TORCH_SRC_DIR}/csrc/jit/mobile/module.cpp
${TORCH_SRC_DIR}/csrc/jit/mobile/register_mobile_ops.cpp
${TORCH_SRC_DIR}/csrc/jit/mobile/register_mobile_autograd.cpp
${TORCH_SRC_DIR}/csrc/jit/mobile/interpreter.cpp
)
list (APPEND TORCH_SRCS ${MOBILE_SRCS})