Fix building Horovod when MXNet is not installed (horovod#2334)

Signed-off-by: Travis Addair <[email protected]>
adangood · Sep 25, 2020 · ee0cdf5 · ee0cdf5
1 parent 4099c2b
commit ee0cdf5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cmake/Modules/FindMxnet.cmake b/cmake/Modules/FindMxnet.cmake
@@ -52,6 +52,9 @@ if (LEN EQUAL "5")
 endif()
 include(FindPackageHandleStandardArgs)
 find_package_handle_standard_args(Mxnet REQUIRED_VARS Mxnet_LIBRARIES VERSION_VAR Mxnet_VERSION)
+if(NOT MXNET_FOUND)
+    return()
+endif()
 
 
 execute_process(COMMAND ${PY_EXE} -c "import mxnet; print(mxnet.library.compiled_with_gcc_cxx11_abi() if hasattr(mxnet, 'library') and hasattr(mxnet.library, 'compiled_with_gcc_cxx11_abi') else 1)"