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) ...@@ -38,7 +38,7 @@ target_link_libraries(caffe2_detectron_custom_ops caffe2_library)
install(TARGETS caffe2_detectron_custom_ops DESTINATION lib) install(TARGETS caffe2_detectron_custom_ops DESTINATION lib)
# Install custom GPU ops lib, if gpu is present. # 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): # Additional -I prefix is required for CMake versions before commit (< 3.7):
# https://github.com/Kitware/CMake/commit/7ded655f7ba82ea72a82d0555449f2df5ef38594 # https://github.com/Kitware/CMake/commit/7ded655f7ba82ea72a82d0555449f2df5ef38594
list(APPEND CUDA_INCLUDE_DIRS -I${CAFFE2_INCLUDE_DIRS}) list(APPEND CUDA_INCLUDE_DIRS -I${CAFFE2_INCLUDE_DIRS})
......
...@@ -11,9 +11,11 @@ function (detectron_print_config_summary) ...@@ -11,9 +11,11 @@ function (detectron_print_config_summary)
message(STATUS " CXX flags : ${CMAKE_CXX_FLAGS}") message(STATUS " CXX flags : ${CMAKE_CXX_FLAGS}")
message(STATUS " Caffe2 version : ${CAFFE2_VERSION}") message(STATUS " Caffe2 version : ${CAFFE2_VERSION}")
message(STATUS " Caffe2 include path : ${CAFFE2_INCLUDE_DIRS}") 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 " CUDA version : ${CUDA_VERSION}")
message(STATUS " CuDNN version : ${CUDNN_VERSION}") message(STATUS " CuDNN version : ${CUDNN_VERSION}")
else()
message(STATUS " Caffe2 found CUDA : False")
endif() endif()
endfunction() 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