Skip to content

GitLab

Unverified Commit 460bb42d authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

fix build problems (#2594)

parent 1f6eba9e
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
src/array/cpu/spmat_op_impl_coo.cc
View file @ 460bb42d
...@@ -3,7 +3,7 @@ ...@@ -3,7 +3,7 @@
* \file array/cpu/spmat_op_impl.cc * \file array/cpu/spmat_op_impl.cc
* \brief CPU implementation of COO sparse matrix operators * \brief CPU implementation of COO sparse matrix operators
*/ */
#include <omp.h> #include <dmlc/omp.h>
#include <vector> #include <vector>
#include <unordered_set> #include <unordered_set>
#include <unordered_map> #include <unordered_map>
......
src/runtime/cuda/cuda_common.h
View file @ 460bb42d
...@@ -93,6 +93,7 @@ struct cuda_dtype<double> { ...@@ -93,6 +93,7 @@ struct cuda_dtype<double> {
static constexpr cudaDataType_t value = CUDA_R_64F; static constexpr cudaDataType_t value = CUDA_R_64F;
}; };
#if CUDART_VERSION >= 11000
/* /*
* \brief Cast index data type to cusparseIndexType_t. * \brief Cast index data type to cusparseIndexType_t.
*/ */
...@@ -110,6 +111,7 @@ template <> ...@@ -110,6 +111,7 @@ template <>
struct cusparse_idtype<int64_t> { struct cusparse_idtype<int64_t> {
static constexpr cusparseIndexType_t value = CUSPARSE_INDEX_64I; static constexpr cusparseIndexType_t value = CUSPARSE_INDEX_64I;
}; };
#endif
/*! \brief Thread local workspace */ /*! \brief Thread local workspace */
class CUDAThreadEntry { class CUDAThreadEntry {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment