commit 17ea11259a6ce6401ac9737c48a59ba9fa3546aa
author Taylor Robie <taylorrobie@fb.com> Tue Dec 01 19:56:13 2020 -0800
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> Tue Dec 01 20:03:11 2020 -0800
tree 7d03a5782443356e83664364195ed5c6640c7c9c
parent 07f038aa9d75bb0befb7fdf4e22b2875cdb0bb49

Rework compat bindings. (#47863)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/47863

Test Plan: Imported from OSS

Reviewed By: ngimel

Differential Revision: D25199261

Pulled By: robieta

fbshipit-source-id: 0a4a0409ddb75c1bf66cd31d67b55080227b1679

setup.py

diff --git a/setup.py b/setup.py
index fd777b1..4ff3ef4 100644
--- a/setup.py
+++ b/setup.py
@@ -327,8 +327,16 @@
 
     # Use copies instead of symbolic files.
     # Windows has very poor support for them.
-    sym_files = ['tools/shared/_utils_internal.py']
-    orig_files = ['torch/_utils_internal.py']
+    sym_files = [
+        'tools/shared/_utils_internal.py',
+        'torch/utils/benchmark/utils/valgrind_wrapper/callgrind.h',
+        'torch/utils/benchmark/utils/valgrind_wrapper/valgrind.h',
+    ]
+    orig_files = [
+        'torch/_utils_internal.py',
+        'third_party/valgrind-headers/callgrind.h',
+        'third_party/valgrind-headers/valgrind.h',
+    ]
     for sym_file, orig_file in zip(sym_files, orig_files):
         same = False
         if os.path.exists(sym_file):
@@ -907,6 +915,8 @@
                 'share/cmake/Gloo/*.cmake',
                 'share/cmake/Tensorpipe/*.cmake',
                 'share/cmake/Torch/*.cmake',
+                'utils/benchmark/utils/valgrind_wrapper/*.cpp',
+                'utils/benchmark/utils/valgrind_wrapper/*.h',
             ],
             'caffe2': [
                 'python/serialized_test/data/operator_test/*.zip',