explicitly include STUBS MPI headers for serial compile since we seem to pick... (6be36196) · Commits · 郑智淋 / lammps

cmake/Modules/Packages/GPU.cmake

+9 −0

Original line number	Diff line number	Diff line
		@@ -99,6 +99,9 @@ if(GPU_API STREQUAL "CUDA")
		add_library(gpu STATIC ${GPU_LIB_SOURCES} ${GPU_LIB_CUDPP_SOURCES} ${GPU_OBJS})
		target_link_libraries(gpu PRIVATE ${CUDA_LIBRARIES} ${CUDA_CUDA_LIBRARY})
		target_include_directories(gpu PRIVATE ${LAMMPS_LIB_BINARY_DIR}/gpu ${CUDA_INCLUDE_DIRS})
		if(NOT BUILD_MPI)
		target_include_directories(gpu PRIVATE ${LAMMPS_SOURCE_DIR}/STUBS)
		endif()
		target_compile_definitions(gpu PRIVATE -D_${GPU_PREC_SETTING} -DMPI_GERYON -DUCL_NO_EXIT ${GPU_CUDA_MPS_FLAGS})
		if(CUDPP_OPT)
		target_include_directories(gpu PRIVATE ${LAMMPS_LIB_SOURCE_DIR}/gpu/cudpp_mini)
		@@ -177,6 +180,9 @@ elseif(GPU_API STREQUAL "OPENCL")
		add_library(gpu STATIC ${GPU_LIB_SOURCES})
		target_link_libraries(gpu PRIVATE OpenCL::OpenCL)
		target_include_directories(gpu PRIVATE ${CMAKE_CURRENT_BINARY_DIR}/gpu)
		if(NOT BUILD_MPI)
		target_include_directories(gpu PRIVATE ${LAMMPS_SOURCE_DIR}/STUBS)
		endif()
		target_compile_definitions(gpu PRIVATE -D_${GPU_PREC_SETTING} -D${OCL_TUNE}_OCL -DMPI_GERYON -DUCL_NO_EXIT)
		target_compile_definitions(gpu PRIVATE -DUSE_OPENCL)

		@@ -280,6 +286,9 @@ elseif(GPU_API STREQUAL "HIP")

		hip_add_library(gpu STATIC ${GPU_LIB_SOURCES})
		target_include_directories(gpu PRIVATE ${LAMMPS_LIB_BINARY_DIR}/gpu)
		if(NOT BUILD_MPI)
		target_include_directories(gpu PRIVATE ${LAMMPS_SOURCE_DIR}/STUBS)
		endif()
		target_compile_definitions(gpu PRIVATE -D_${GPU_PREC_SETTING} -DMPI_GERYON -DUCL_NO_EXIT)
		target_compile_definitions(gpu PRIVATE -DUSE_HIP)