saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use CMake built-in function to compare NCCL version (#5118)

Browse source 
* Use CMake built-in function to compare version

* Address comment
This commit is contained in:
Wei-Sheng Chin 2020-09-10 15:59:47 -07:00 committed by GitHub
parent c5d4ae0401
commit 5618b9dddc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
cmake/CMakeLists.txt
View file

@ -1186,13 +1186,10 @@ if (onnxruntime_ENABLE_TRAINING)
message(STATUS "NCCL_MINOR_VERSION: ${NCCL_MINOR_VERSION}")
endif()
if (${NCCL_MAJOR_VERSION} GREATER 2)
add_definitions(-DUSE_NCCL_P2P=1)
else()
if(${NCCL_MAJOR_VERSION} EQUAL 2)
if(${NCCL_MINOR_VERSION} GREATER 7)
add_definitions(-DUSE_NCCL_P2P=1)
endif()
if (NCCL_MAJOR_VERSION_DEFINED AND NCCL_MINOR_VERSION_DEFINED)
if ("${NCCL_MAJOR_VERSION}.${NCCL_MINOR_VERSION}" VERSION_GREATER_EQUAL "2.7")
add_definitions(-DUSE_NCCL_P2P=1)
message( STATUS "NCCL P2P is enabled for supporting ncclSend and ncclRecv." )
endif()
endif()