1 files changed, 4 insertions, 0 deletions
diff --git a/setup.py b/setup.py
index a0066083a3..9b3d45d09d 100644
--- a/setup.py
+++ b/setup.py
@@ -864,11 +864,15 @@ if __name__ == '__main__':
                 'lib/torch_shm_manager',
                 'lib/*.h',
                 'lib/include/ATen/*.h',
+                'lib/include/ATen/cuda/*.cuh',
+                'lib/include/ATen/cudnn/*.h',
+                'lib/include/ATen/cuda/detail/*.cuh',
                 'lib/include/pybind11/*.h',
                 'lib/include/pybind11/detail/*.h',
                 'lib/include/TH/*.h',
                 'lib/include/TH/generic/*.h',
                 'lib/include/THC/*.h',
+                'lib/include/THC/*.cuh',
                 'lib/include/THC/generic/*.h',
                 'lib/include/torch/csrc/*.h',
                 'lib/include/torch/csrc/autograd/*.h',