Grab changes

carbonrobotics · Dec 28, 2023 · 662e513 · 662e513
1 parent 93a2f25
commit 662e513
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/horovod/torch/CMakeLists.txt b/horovod/torch/CMakeLists.txt
@@ -56,7 +56,9 @@ endif()
 parse_version(${Pytorch_VERSION} VERSION_DEC)
 add_definitions(-DTORCH_VERSION=${VERSION_DEC} -DTORCH_API_INCLUDE_EXTENSION_H=1)
 set(Pytorch_CXX11 ${Pytorch_CXX11} PARENT_SCOPE)
-if(NOT Pytorch_VERSION VERSION_LESS "1.3.0")
+if(NOT Pytorch_VERSION VERSION_LESS "2.1.0")
+    set(CMAKE_CXX_STANDARD 17)
+elseif(NOT Pytorch_VERSION VERSION_LESS "1.3.0")
     set(CMAKE_CXX_STANDARD 14)
 endif()