From: Dmytro Dzhulgakov <dzhulgakov@fb.com>
Date: Thu, 14 Feb 2019 06:53:56 +0000 (-0800)
Subject: Add more headers to setup.py to make pytorch/benchmark work (#16890)
X-Git-Tag: accepted/tizen/6.5/unified/20211028.231830~1290
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=5a26579e27aab8ab13860ff25f8c370ab2a60f91;p=platform%2Fupstream%2Fpytorch.git

Add more headers to setup.py to make pytorch/benchmark work (#16890)

Summary:
Since we don't do tmp_install any more it's better to include all necessary headers.

cc kostmo for better suggestions of how to list all headers here
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16890

Differential Revision: D14079848

Pulled By: dzhulgakov

fbshipit-source-id: 4522c80d05e5d91f99f6700cde46cac559330d28
---

diff --git a/setup.py b/setup.py
index c02a381..d21567b 100644
--- a/setup.py
+++ b/setup.py
@@ -792,12 +792,15 @@ if __name__ == '__main__':
                 'include/torch/csrc/api/include/torch/optim/*.h',
                 'include/torch/csrc/api/include/torch/serialize/*.h',
                 'include/torch/csrc/autograd/*.h',
+                'include/torch/csrc/autograd/functions/*.h',
                 'include/torch/csrc/autograd/generated/*.h',
+                'include/torch/csrc/autograd/utils/*.h',
                 'include/torch/csrc/cuda/*.h',
                 'include/torch/csrc/jit/*.h',
                 'include/torch/csrc/jit/generated/*.h',
                 'include/torch/csrc/jit/passes/*.h',
                 'include/torch/csrc/jit/script/*.h',
+                'include/torch/csrc/onnx/*.h',
                 'include/torch/csrc/utils/*.h',
                 'include/pybind11/*.h',
                 'include/pybind11/detail/*.h',