Passing Array as argument in SpMMCsr() of /src/array/cuda/spmm.cu in DGL

Questions
#1

Hello,
I am partitioning the DGL graph using dgl.distributed.partition_graph and storing partitioned nodes in array. Now I want to pass this array in SpMMCsr() for SpMM operation in /src/array/cuda/spmm.cu. How to pass this array as an argument in SpMMCsr()?

#2

Hi, I’d recommend you to use the new sparse matrix API. You can first create the sparse matrix using the from_csr function and then use the spmm operator. The operator will invoke SpMMCsr internally.

#3

template void SpMMCsr<kDGLCUDA, int32_t, 16>(
const std::string& op, const std::string& reduce,
const BcastOff& bcast, const CSRMatrix& csr,
NDArray ufeat, NDArray efeat, NDArray out, std::vector out_aux). I want to use array in this template. It is giving error could you please tell me how to add array here

#4

This topic was automatically closed 30 days after the last reply. New replies are no longer allowed.