spack/var/spack/repos/builtin/packages/py-torch/sleef.patch
Adam J. Stewart 4f9b539644
py-torch: overhaul package (#24294)
* py-torch: patch no longer needed on master

* Overhaul PyTorch package

* py-torch: add v1.9.0

* Change defaults on macOS

* Submodules still needed...

* Add ONNX dependency

* System libs don't work for many submodules

* Silence CMake warning

* Add conflict for +cuda+rocm

* Add more deps

* Add more BLAS options

* Disable some broken system libs options

* Add patches to build older versions

* +mkldnn requires mkl

* Fix BLAS settings
2021-06-30 12:38:53 -04:00

13 lines
661 B
Diff

diff --git a/caffe2/CMakeLists.txt b/caffe2/CMakeLists.txt
index 8025a7de3c..2e5cdbb5c9 100644
--- a/caffe2/CMakeLists.txt
+++ b/caffe2/CMakeLists.txt
@@ -1232,6 +1232,7 @@ if (BUILD_TEST)
add_executable(${test_name} "${test_src}")
target_link_libraries(${test_name} ${Caffe2_MAIN_LIBS} gtest_main)
target_include_directories(${test_name} PRIVATE $<INSTALL_INTERFACE:include>)
+ target_include_directories(${test_name} PRIVATE $<BUILD_INTERFACE:${CMAKE_BINARY_DIR}/include>)
target_include_directories(${test_name} PRIVATE ${Caffe2_CPU_INCLUDE})
add_test(NAME ${test_name} COMMAND $<TARGET_FILE:${test_name}>)
if (INSTALL_TEST)