Commit 8170b25b authored by Ilija Radosavovic's avatar Ilija Radosavovic Committed by Facebook Github Bot

Update CMake found CUDA summary

Reviewed By: rbgirshick

Differential Revision: D8614301

fbshipit-source-id: 384df17bebd0a0186a0867cf16a7a7dfea5646ae
parent cd447c77
......@@ -38,7 +38,7 @@ target_link_libraries(caffe2_detectron_custom_ops caffe2_library)
install(TARGETS caffe2_detectron_custom_ops DESTINATION lib)
# Install custom GPU ops lib, if gpu is present.
if ("${CAFFE2_USE_CUDA}" OR "${CAFFE2_FOUND_CUDA}")
if (CAFFE2_USE_CUDA OR CAFFE2_FOUND_CUDA)
# Additional -I prefix is required for CMake versions before commit (< 3.7):
# https://github.com/Kitware/CMake/commit/7ded655f7ba82ea72a82d0555449f2df5ef38594
list(APPEND CUDA_INCLUDE_DIRS -I${CAFFE2_INCLUDE_DIRS})
......
......@@ -11,9 +11,11 @@ function (detectron_print_config_summary)
message(STATUS " CXX flags : ${CMAKE_CXX_FLAGS}")
message(STATUS " Caffe2 version : ${CAFFE2_VERSION}")
message(STATUS " Caffe2 include path : ${CAFFE2_INCLUDE_DIRS}")
message(STATUS " Caffe2 found CUDA : ${CAFFE2_FOUND_CUDA}")
if ("${CAFFE2_USE_CUDA}" OR "${CAFFE2_FOUND_CUDA}")
if (CAFFE2_USE_CUDA OR CAFFE2_FOUND_CUDA)
message(STATUS " Caffe2 found CUDA : True")
message(STATUS " CUDA version : ${CUDA_VERSION}")
message(STATUS " CuDNN version : ${CUDNN_VERSION}")
else()
message(STATUS " Caffe2 found CUDA : False")
endif()
endfunction()
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment