This commit is contained in:
Ruipeng Li 2022-06-09 00:11:31 -07:00
parent f6edbf7631
commit 26d53aacce
2 changed files with 5 additions and 0 deletions

View File

@ -6,6 +6,8 @@
******************************************************************************/
#include "seq_mv.h"
#if defined(HYPRE_USING_CUDA) || defined(HYPRE_USING_HIP)
#define HYPRE_SPGEMM_DEVICE_USE_DSHMEM
#include <csr_spgemm_device_numer.h>
@ -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) */

View File

@ -5,6 +5,7 @@
* SPDX-License-Identifier: (Apache-2.0 OR MIT)
******************************************************************************/
#include <math.h>
#include "_hypre_onedpl.hpp"
#include "_hypre_utilities.h"
#include "_hypre_utilities.hpp"