From 37890610b02fae2bd4b4eff3fd3747a3dc0ffaad Mon Sep 17 00:00:00 2001 From: Tri Dao Date: Thu, 21 Feb 2019 07:34:27 -0800 Subject: [PATCH] Include vec256 headers in setup.py (#17220) Summary: Fix #16650. Headers such as `ATen/cpu/vml.h` contain `#include ` for example, but these vec256 headers aren't included, due to commit e4c0bb1. Pull Request resolved: https://github.com/pytorch/pytorch/pull/17220 Differential Revision: D14165695 Pulled By: ezyang fbshipit-source-id: 27b2aa2a734b3719ca4af0565f79623b64b2620f --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index 3318bbb..7dd510a 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', -- 2.7.4