1- cmake_minimum_required (VERSION 3.18...3.27 )
1+ cmake_minimum_required (VERSION 3.18)
22project (${SKBUILD_PROJECT_NAME} VERSION ${SKBUILD_PROJECT_VERSION} LANGUAGES CXX)
33
44# General settings
@@ -21,37 +21,28 @@ if(CMAKE_CUDA_COMPILER)
2121 message (STATUS "CMAKE_CUDA_COMPILER = ${CMAKE_CUDA_COMPILER} " )
2222 message (STATUS "CMAKE_CUDA_COMPILER_ID = ${CMAKE_CUDA_COMPILER_ID} " )
2323 message (STATUS "CMAKE_CUDA_COMPILER_VERSION = ${CUDA_V} " )
24-
2524else ()
26- message (STATUS "NO CUDA INSTALLATION FOUND, INSTALLING CPU VERSION ONLY!" )
27- # execute_process(COMMAND ${Python_EXECUTABLE} -m pip uninstall -y torch)
28- # execute_process(COMMAND ${Python_EXECUTABLE} -m pip install -v torch --extra-index-url https://download.pytorch.org/whl/cpu)
25+ message (WARNING "NO CUDA INSTALLATION FOUND, TRYING TO INSTALL CPU VERSION ONLY!" )
2926endif ()
3027
3128set (CMAKE_CXX_STANDARD 17)
3229set (CMAKE_CXX_STANDARD_REQUIRED ON )
3330set (CMAKE_POSITION_INDEPENDENT_CODE ON )
3431
3532if (MSVC )
36- add_definitions (-DNOMINMAX)
33+ add_compile_options ("$<$<COMPILE_LANGUAGE:CXX>:/utf-8>" )
34+ add_compile_options ("$<$<COMPILE_LANGUAGE:CUDA>:-Xcompiler=/utf-8>" )
35+ add_definitions (-DNOMINMAX -D_CRT_SECURE_NO_WARNINGS)
3736endif ()
3837
3938# Find packages
40- find_package (Python REQUIRED COMPONENTS Interpreter Development)
39+ find_package (Python COMPONENTS Interpreter Development.Module REQUIRED )
4140find_package (pybind11 CONFIG REQUIRED)
42-
43- # OpenMP
4441find_package (OpenMP)
45- if (OPENMP_FOUND)
46- set (CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS} " )
47- set (CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OpenMP_CXX_FLAGS} " )
48- endif ()
4942
50- # PyTorch settings
51- find_package (Torch REQUIRED)
43+ find_package (Torch CONFIG REQUIRED)
5244find_library (TORCH_PYTHON_LIBRARY torch_python PATH "${TORCH_INSTALL_PREFIX} /lib" )
5345set (CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS} " )
5446
5547add_subdirectory (cxx)
56-
5748install (TARGETS ${BUILD_TARGET} LIBRARY DESTINATION .)
0 commit comments