diff --git a/src/seq_mv/csr_spgemm_device_numer10.c b/src/seq_mv/csr_spgemm_device_numer10.c index 229fff518..0db77459e 100644 --- a/src/seq_mv/csr_spgemm_device_numer10.c +++ b/src/seq_mv/csr_spgemm_device_numer10.c @@ -6,6 +6,8 @@ ******************************************************************************/ #include "seq_mv.h" +#if defined(HYPRE_USING_CUDA) || defined(HYPRE_USING_HIP) + #define HYPRE_SPGEMM_DEVICE_USE_DSHMEM #include @@ -24,3 +26,5 @@ hypre_spgemm_numerical_max_num_blocks < HYPRE_SPGEMM_NUMER_HASH_SIZE * 32, HYPRE_SPGEMM_BASE_GROUP_SIZE * 32 > ( HYPRE_Int multiProcessorCount, HYPRE_Int *num_blocks_ptr, HYPRE_Int *block_size_ptr ); +#endif /* HYPRE_USING_CUDA || defined(HYPRE_USING_HIP) */ + diff --git a/src/utilities/device_utils.c b/src/utilities/device_utils.c index 07ee31fcd..d4e7b5728 100644 --- a/src/utilities/device_utils.c +++ b/src/utilities/device_utils.c @@ -5,6 +5,7 @@ * SPDX-License-Identifier: (Apache-2.0 OR MIT) ******************************************************************************/ +#include #include "_hypre_onedpl.hpp" #include "_hypre_utilities.h" #include "_hypre_utilities.hpp"