Skip to content

Commit 2b57ef9

Browse files
fmassaRob Kunkle
authored and
Rob Kunkle
committed
Install hpp headers for CPP Extensions (pytorch#9182)
Summary: With the Cppzation of a few files in `TH`/`THC`, the CPP extensions got broken whenever the user uses feature from `THC` in their files, when pytorch is installed via `python setup.py install`. This addresses issues such as ``` /home/me/.conda/envs/pytorch/lib/python3.6/site-packages/torch/lib/include/THC/THCDeviceTensorUtils.cuh:5:25: fatal error: THCTensor.hpp: No such file or directory ``` Closes pytorch#9182 Reviewed By: soumith Differential Revision: D8734581 Pulled By: fmassa fbshipit-source-id: 2a1138f208592eaccb01fcdb805a6b369d7a497a
1 parent 297b925 commit 2b57ef9

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

setup.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1060,9 +1060,9 @@ def make_relative_rpath(path):
10601060
'lib/include/ATen/cuda/detail/*.cuh',
10611061
'lib/include/pybind11/*.h',
10621062
'lib/include/pybind11/detail/*.h',
1063-
'lib/include/TH/*.h',
1064-
'lib/include/TH/generic/*.h',
1065-
'lib/include/THC/*.h',
1063+
'lib/include/TH/*.h*',
1064+
'lib/include/TH/generic/*.h*',
1065+
'lib/include/THC/*.h*',
10661066
'lib/include/THC/*.cuh',
10671067
'lib/include/THC/generic/*.h',
10681068
'lib/include/THCUNN/*.cuh',

0 commit comments

Comments
 (0)