saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix BUILD_CAFFE2_OPS

Browse source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/16783

Differential Revision: D13965061

Pulled By: zdevito

fbshipit-source-id: 6fe710ca51e2f338873b56f23256668ca3fe2032
This commit is contained in:
Zachary DeVito 2019-02-05 22:36:48 -08:00 committed by Facebook Github Bot
parent ab035d01e3
commit 255136fc1d
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
torch/CMakeLists.txt
View file

@ -135,7 +135,6 @@ set(TORCH_SRCS
${TORCH_SRC_DIR}/csrc/jit/ir.cpp
${TORCH_SRC_DIR}/csrc/jit/operator.cpp
${TORCH_SRC_DIR}/csrc/jit/caffe2_operator.cpp
${TORCH_SRC_DIR}/csrc/jit/register_caffe2_ops.cpp
${TORCH_SRC_DIR}/csrc/jit/symbolic_script.cpp
${TORCH_SRC_DIR}/csrc/jit/passes/alias_analysis.cpp
${TORCH_SRC_DIR}/csrc/jit/passes/batch_mm.cpp
@ -191,6 +190,11 @@ set(TORCH_SRCS
${TORCH_ROOT}/test/cpp/jit/no-gtest.cpp
)
if (BUILD_CAFFE2_OPS)
list(APPEND TORCH_SRCS
${TORCH_SRC_DIR}/csrc/jit/register_caffe2_ops.cpp)
endif()
if (WIN32)
list(APPEND TORCH_SRCS
${TORCH_SRC_DIR}/csrc/jit/fuser/cpu/dynamic_library_win.cpp