Skip to content

Commit

Permalink
continuing heffte cuda
Browse files Browse the repository at this point in the history
  • Loading branch information
djdunning committed Aug 1, 2024
1 parent 884ef87 commit b486126
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions .conda/heffte/cuda/build.sh
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ fi

source "$RECIPE_DIR/../../cross-compile-setup.sh"

#export OMPI_CXX=nvcc
export OMPI_CXX=nvcc
# Only do this for cross compiling
#if [ "$PLATFORM" != "linux-64" ] ; then
# export NVCC_PREPEND_FLAGS="-ccbin $CXX";
Expand All @@ -24,14 +24,14 @@ export LIBRARIES="$LIBRARIES \"-L$PREFIX/lib\""
# -D CMAKE_CUDA_HOST_LINK_LAUNCHER=$CXX \
# -D MPI_C_COMPILER="$BUILD_PREFIX/bin/mpicc" \
# -D MPI_CXX_COMPILER="$BUILD_PREFIX/bin/mpicxx" \
# -D CMAKE_CUDA_COMPILER=nvcc \
cmake -D CMAKE_BUILD_TYPE:STRING=RELEASE \
-D BUILD_SHARED_LIBS=ON \
-D CMAKE_INSTALL_PREFIX:PATH=$PREFIX \
-D CMAKE_CXX_STANDARD:STRING=17 \
-D Heffte_ENABLE_CUDA=ON \
-D Heffte_DISABLE_GPU_AWARE_MPI=ON \
-D CMAKE_CXX_FLAGS="$PATCHED_CXXFLAGS" \
-D CMAKE_CUDA_COMPILER=nvcc \
$CMAKE_ARGS \
-S $SRC_DIR \

Expand Down

0 comments on commit b486126

Please sign in to comment.