[Caffe2] add the missed header files

add the missed header files when caffe2 is installed

Signed-off-by: Hyoung Joo Ahn <hello.ahn@samsung.com>

1 files changed, 2 insertions, 0 deletions

diff --git a/setup.py b/setup.py
index b6a96d7926..8808030f67 100644
--- a/setup.py
+++ b/setup.py
@@ -754,6 +754,8 @@ if __name__ == '__main__':
                 'include/ATen/cudnn/*.h',
                 'include/ATen/detail/*.h',
                 'include/caffe2/utils/*.h',
+                'include/caffe2/utils/math/*.h',
+                'include/caffe2/utils/threadpool/*.h',
                 'include/c10/*.h',
                 'include/c10/macros/*.h',
                 'include/c10/core/*.h',