diff --git a/setup.py b/setup.py
index 3318bbb07e11..7dd510a0cb24 100644
--- a/setup.py
+++ b/setup.py
@@ -755,6 +755,7 @@ if __name__ == '__main__':
                 'lib/*.h',
                 'include/ATen/*.h',
                 'include/ATen/cpu/*.h',
+                'include/ATen/cpu/vec256/*.h',
                 'include/ATen/core/*.h',
                 'include/ATen/cuda/*.cuh',
                 'include/ATen/cuda/*.h',