fdurastante
/
psblas3
mirror of https://github.com/sfilippone/psblas3.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add cuda building and compilation. There is still some hardcoded stuff

Browse Source
pull/28/head
gnumlab 1 year ago
parent 1b667c2c1e
commit f953c52591
1 changed files with 62 additions and 21 deletions
Show Stats Download Patch File Download Diff File

83
CMakeLists.txt
Unescape Escape View File

@ -651,7 +651,48 @@ if(MPI_FOUND)
endif() endif()
endif() endif()
# Optionally check for CUDA requirement
option(PSB_BUILD_CUDA "Build CUDA code" OFF)
if(PSB_BUILD_CUDA)
if(NOT DEFINED PSB_CUDA_PATH)
set(PSB_CUDA_PATH "/opt/cuda/12.8")
endif()
# Include the CMakeLists for the cbind library
include(${CMAKE_CURRENT_LIST_DIR}/cuda/CMakeLists.txt)
include_directories("${PSB_CUDA_PATH}/include")
message(STATUS "${PSB_CUDA_PATH}")
# find_package(CUDA REQUIRED)
enable_language(CUDA)
# Check for CUDA version
set(PSB_CUDA_VERSION 12800)
if(PSB_CUDA_VERSION)
message(STATUS "CUDA version: ${PSB_CUDA_VERSION}")
# Define macros for CUDA version
# add_definitions(-DPSB_HAVE_CUDA)
# add_definitions(-DPSB_CUDA_VERSION=${PSB_CUDA_VERSION})
math(EXPR PSB_CUDA_SHORT_VERSION "${PSB_CUDA_VERSION} / 1000")
# add_definitions(-DPSB_CUDA_SHORT_VERSION=${PSB_CUDA_SHORT_VERSION})
set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -DPSB_HAVE_CUDA")
set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -DPSB_CUDA_VERSION=${PSB_CUDA_VERSION}")
set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -DPSB_CUDA_SHORT_VERSION=${PSB_CUDA_SHORT_VERSION}")
set(CHAVECUDA "#define PSB_HAVE_CUDA")
set(CSHORTVCUDA "#define PSB_CUDA_SHORT_VERSION ${PSB_CUDA_SHORT_VERSION}")
set(CVERSIONCUDA "#define PSB_CUDA_VERSION ${PSB_CUDA_VERSION}")
else()
message(FATAL_ERROR "CUDA version not found!")
endif()
endif()
#------------------------------------------ #------------------------------------------
# Configure the psb_config.h file # Configure the psb_config.h file
@ -955,46 +996,46 @@ message(STATUS "Copied .h files to ${CMAKE_BINARY_DIR}/include")
# Optionally check for CUDA requirement # Optionally check for CUDA requirement
option(PSB_BUILD_CUDA "Build CUDA code" OFF) #option(PSB_BUILD_CUDA "Build CUDA code" OFF)
if(PSB_BUILD_CUDA) if(PSB_BUILD_CUDA)
if(NOT DEFINED PSB_CUDA_PATH) # if(NOT DEFINED PSB_CUDA_PATH)
set(PSB_CUDA_PATH "/opt/cuda/12.8") # set(PSB_CUDA_PATH "/opt/cuda/12.8")
endif() #endif()
# Include the CMakeLists for the cbind library # Include the CMakeLists for the cbind library
include(${CMAKE_CURRENT_LIST_DIR}/cuda/CMakeLists.txt) #include(${CMAKE_CURRENT_LIST_DIR}/cuda/CMakeLists.txt)
include_directories("${PSB_CUDA_PATH}/include") #include_directories("${PSB_CUDA_PATH}/include")
message(STATUS "${PSB_CUDA_PATH}") #message(STATUS "${PSB_CUDA_PATH}")
# find_package(CUDA REQUIRED) # find_package(CUDA REQUIRED)
enable_language(CUDA) #enable_language(CUDA)
# Check for CUDA version # Check for CUDA version
set(PSB_CUDA_VERSION 12800) #set(PSB_CUDA_VERSION 12800)
if(PSB_CUDA_VERSION) #if(PSB_CUDA_VERSION)
message(STATUS "CUDA version: ${PSB_CUDA_VERSION}") # message(STATUS "CUDA version: ${PSB_CUDA_VERSION}")
# Define macros for CUDA version # Define macros for CUDA version
# add_definitions(-DPSB_HAVE_CUDA) # add_definitions(-DPSB_HAVE_CUDA)
# add_definitions(-DPSB_CUDA_VERSION=${PSB_CUDA_VERSION}) # add_definitions(-DPSB_CUDA_VERSION=${PSB_CUDA_VERSION})
math(EXPR PSB_CUDA_SHORT_VERSION "${PSB_CUDA_VERSION} / 1000") #math(EXPR PSB_CUDA_SHORT_VERSION "${PSB_CUDA_VERSION} / 1000")
# add_definitions(-DPSB_CUDA_SHORT_VERSION=${PSB_CUDA_SHORT_VERSION}) # add_definitions(-DPSB_CUDA_SHORT_VERSION=${PSB_CUDA_SHORT_VERSION})
set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -DPSB_HAVE_CUDA") #set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -DPSB_HAVE_CUDA")
set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -DPSB_CUDA_VERSION=${PSB_CUDA_VERSION}") #set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -DPSB_CUDA_VERSION=${PSB_CUDA_VERSION}")
set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -DPSB_CUDA_SHORT_VERSION=${PSB_CUDA_SHORT_VERSION}") #set(CMAKE_Fortran_FLAGS "${CMAKE_Fortran_FLAGS} -DPSB_CUDA_SHORT_VERSION=${PSB_CUDA_SHORT_VERSION}")
set(CHAVECUDA "#define PSB_HAVE_CUDA") #set(CHAVECUDA "#define PSB_HAVE_CUDA")
set(CSHORTVCUDA "#define PSB_CUDA_SHORT_VERSION ${PSB_CUDA_SHORT_VERSION}") #set(CSHORTVCUDA "#define PSB_CUDA_SHORT_VERSION ${PSB_CUDA_SHORT_VERSION}")
set(CVERSIONCUDA "#define PSB_CUDA_VERSION ${PSB_CUDA_VERSION}") #set(CVERSIONCUDA "#define PSB_CUDA_VERSION ${PSB_CUDA_VERSION}")
else() #else()
message(FATAL_ERROR "CUDA version not found!") #message(FATAL_ERROR "CUDA version not found!")
endif() #endif()

Write Preview
Loading…
Cancel
Save