"docker/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "66956356968b4849a98c2a86d58a5dd588d638c8"
Unverified Commit 460bb42d authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

fix build problems (#2594)

parent 1f6eba9e
......@@ -3,7 +3,7 @@
* \file array/cpu/spmat_op_impl.cc
* \brief CPU implementation of COO sparse matrix operators
*/
#include <omp.h>
#include <dmlc/omp.h>
#include <vector>
#include <unordered_set>
#include <unordered_map>
......
......@@ -93,6 +93,7 @@ struct cuda_dtype<double> {
static constexpr cudaDataType_t value = CUDA_R_64F;
};
#if CUDART_VERSION >= 11000
/*
* \brief Cast index data type to cusparseIndexType_t.
*/
......@@ -110,6 +111,7 @@ template <>
struct cusparse_idtype<int64_t> {
static constexpr cusparseIndexType_t value = CUSPARSE_INDEX_64I;
};
#endif
/*! \brief Thread local workspace */
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