diff --git a/.conda/evpfft/cuda/build.sh b/.conda/evpfft/cuda/build.sh index 199b3eb46..eb92a79a7 100755 --- a/.conda/evpfft/cuda/build.sh +++ b/.conda/evpfft/cuda/build.sh @@ -12,8 +12,12 @@ source "$RECIPE_DIR/../../cross-compile-setup.sh" # ^ Passes $NVCC_WRAPPER_DEFAULT_COMPILER as the host compiler to nvcc # Setting NVCC_WRAPPER_DEFAULT_COMPILER=$GXX enforces that # nvcc gets the correct compiler for the target platform. -export OMPI_CXX=nvcc_wrapper -export NVCC_WRAPPER_DEFAULT_COMPILER=$GXX +#export OMPI_CXX=nvcc_wrapper +#export NVCC_WRAPPER_DEFAULT_COMPILER=$GXX + + #-D CMAKE_CXX_COMPILER=$BUILD_PREFIX/bin/mpicxx \ + #-D CMAKE_C_COMPILER=$BUILD_PREFIX/bin/mpicc \ + #-D CUDAToolkit_ROOT=$PREFIX/bin \ cd src/EVPFFT mkdir build @@ -23,8 +27,5 @@ cmake ../src/ \ -D USE_CUFFT=1 \ $CMAKE_ARGS \ -D CMAKE_CXX_FLAGS="$PATCHED_CXXFLAGS" \ - -D CUDAToolkit_ROOT=$PREFIX/bin \ - -D CMAKE_CXX_COMPILER=$BUILD_PREFIX/bin/mpicxx \ - -D CMAKE_C_COMPILER=$BUILD_PREFIX/bin/mpicc \ -make install \ No newline at end of file +make install diff --git a/.conda/evpfft/cuda/meta.yaml b/.conda/evpfft/cuda/meta.yaml index a4f06ea3f..1169648c0 100644 --- a/.conda/evpfft/cuda/meta.yaml +++ b/.conda/evpfft/cuda/meta.yaml @@ -27,6 +27,7 @@ requirements: - {{ compiler('cxx') }}={{ macos_compiler_version }} # [osx] - cuda-toolkit - kokkos=4.1 + - matar-cuda - openmpi host: - openmpi diff --git a/lib/Elements b/lib/Elements index 19b5cad11..0de655cd2 160000 --- a/lib/Elements +++ b/lib/Elements @@ -1 +1 @@ -Subproject commit 19b5cad11d411a914cf188814afb2bd6180c85b7 +Subproject commit 0de655cd260e60e4f037a78561dddfb9c8757fc7 diff --git a/src/EVPFFT/src/CMakeLists.txt b/src/EVPFFT/src/CMakeLists.txt index 026382e6a..754ae95ba 100644 --- a/src/EVPFFT/src/CMakeLists.txt +++ b/src/EVPFFT/src/CMakeLists.txt @@ -62,13 +62,13 @@ endif() # HAVE_KOKKOS must be defined for MATAR to build Kokkos types add_definitions(-DHAVE_KOKKOS=1) -#if (NOT BUILD_EVPFFT_FIERRO) - #if (USE_CUFFT) - # find_package(CUDAToolkit REQUIRED) - #endif() -# find_package(Kokkos REQUIRED) -#endif() -find_package(Kokkos REQUIRED) +if (NOT BUILD_EVPFFT_FIERRO) + if (USE_CUFFT) + find_package(CUDAToolkit REQUIRED) + endif() + find_package(Kokkos REQUIRED) +endif() +#find_package(Kokkos REQUIRED) find_package(MPI REQUIRED) find_package(Heffte REQUIRED) # HDF5 @@ -136,8 +136,10 @@ elseif (USE_MKL) add_definitions(-DUSE_MKL=1) elseif (USE_CUFFT) add_definitions(-DUSE_CUFFT=1) + add_definitions(-DHAVE_CUDA=1) elseif (USE_ROCFFT) add_definitions(-DUSE_ROCFFT=1) + add_definitions(-DHAVE_HIP=1) else () message(FATAL_ERROR "\nPlease specify what FFT backend to use: -DUSE_FFTW=1, -DUSE_MKL=1, -DUSE_CUFFT=1, or -DUSE_ROCFFT=1\n") endif ()