saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-06 00:03:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Oops, wrong CMakeLists

Browse source
This commit is contained in:
Ryan Hill 2021-05-11 22:50:42 -07:00
parent 4ad3a41ff9
commit f4bb6f5c32
1 changed files with 2 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
cmake/CMakeLists.txt
View file

@ -1599,17 +1599,9 @@ if (onnxruntime_ENABLE_TRAINING)
endif()
endif()
# if (onnxruntime_USE_MPI AND MPI_FOUND)
set(MPI_INCLUDE_DIRS "C:/code/openmpi-4.0.4/ompi/include")
set(NCCL_INCLUDE_DIRS "C:/Code/nccl-1.3.4-1/src")
add_definitions(-DORT_USE_NCCL=1)
set(onnxruntime_USE_NCCL ON)
if (onnxruntime_USE_MPI AND MPI_FOUND)
add_definitions(-DUSE_MPI=1)
# endif()
# if (onnxruntime_USE_MPI AND MPI_FOUND)
#add_definitions(-DUSE_MPI=1)
#endif()
endif()
add_subdirectory(tensorboard EXCLUDE_FROM_ALL)
list(APPEND onnxruntime_EXTERNAL_LIBRARIES tensorboard)