summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorParichay Kapoor <pk.kapoor@samsung.com>2019-06-13 14:16:31 +0900
committerGitHub <noreply@github.com>2019-06-13 14:16:31 +0900
commit84ef2c0efa21a55681214114c4c2128503d735d4 (patch)
tree0ae0a3bc4b4c6373b905664a2cd14219175dadc4
parent1717fb035a348810fe89620faae1d595410cbe2d (diff)
parent1178557c9b167da6b48525d037c2d9a5a75d6835 (diff)
downloadpytorch-84ef2c0efa21a55681214114c4c2128503d735d4.tar.gz
pytorch-84ef2c0efa21a55681214114c4c2128503d735d4.tar.bz2
pytorch-84ef2c0efa21a55681214114c4c2128503d735d4.zip
Merge pull request #1 from helloahn/caffe2_headers
[Caffe2] add the missed header files
-rw-r--r--setup.py2
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',