commit 1a23c9901dbfee295bf5b3dad36e4d3ee7e86366
author Sam Gross <colesbury@gmail.com> Mon Mar 12 15:44:10 2018 -0400
committer GitHub <noreply@github.com> Mon Mar 12 15:44:10 2018 -0400
tree e934c8c8bf78d7f3d037fb8f51fcdfb816561e39
parent 4007dd76e20a4e222fa330a44d38d73aaa70734e

Check that new cpuinfo and tbb submodules exist (#5714)

setup.py

diff --git a/setup.py b/setup.py
index ac19a65..b4941ec 100644
--- a/setup.py
+++ b/setup.py
@@ -252,7 +252,8 @@
                 sys.exit(1)
         check_file(os.path.join(lib_path, "gloo", "CMakeLists.txt"))
         check_file(os.path.join(lib_path, "nanopb", "CMakeLists.txt"))
-        check_file(os.path.join(lib_path, "pybind11", "CMakeLists.txt"))
+        check_file(os.path.join('aten', 'src', 'ATen', 'cpu', 'cpuinfo', 'CMakeLists.txt'))
+        check_file(os.path.join('aten', 'src', 'ATen', 'cpu', 'tbb', 'tbb_remote', 'Makefile'))
 
         check_pydep('yaml', 'pyyaml')
         check_pydep('typing', 'typing')