diff --git a/caffe2/CMakeLists.txt b/caffe2/CMakeLists.txt index 2610ec3963afb..fb32b8d87c341 100644 --- a/caffe2/CMakeLists.txt +++ b/caffe2/CMakeLists.txt @@ -1143,7 +1143,7 @@ endif() install(DIRECTORY "${TORCH_SRC_DIR}/csrc" DESTINATION ${TORCH_INSTALL_INCLUDE_DIR}/torch - FILES_MATCHING PATTERN "*.h") + FILES_MATCHING PATTERN "*.h" PATTERN "*.hpp") install(DIRECTORY "${TORCH_SRC_DIR}/csrc/distributed/c10d" DESTINATION ${TORCH_INSTALL_INCLUDE_DIR} FILES_MATCHING PATTERN "*.h" PATTERN "*.hpp") diff --git a/setup.py b/setup.py index 21e983edfd1e0..410edfefa7e72 100644 --- a/setup.py +++ b/setup.py @@ -1019,7 +1019,8 @@ def print_box(msg): 'include/torch/csrc/autograd/utils/*.h', 'include/torch/csrc/cuda/*.h', 'include/torch/csrc/deploy/*.h', - 'include/torch/csrc/deploy/interpreter/interpreter_impl.h', + 'include/torch/csrc/deploy/interpreter/*.h', + 'include/torch/csrc/deploy/interpreter/*.hpp', 'include/torch/csrc/distributed/c10d/exception.h', 'include/torch/csrc/jit/*.h', 'include/torch/csrc/jit/backends/*.h',