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',