cusparseStatus_t __cusparseXcsrgemm__(cusparseHandle_t handle,
cusparseOperation_t transA, cusparseOperation_t transB,
int m, int n, int k,
const cusparseMatDescr_t descrA, const int nnzA,
const float *csrValA,
const int *csrRowPtrA, const int *csrColIndA,
const cusparseMatDescr_t descrB, const int nnzB,
const float *csrValB,
const int *csrRowPtrB, const int *csrColIndB,
const cusparseMatDescr_t descrC,
float *csrValC,
const int *csrRowPtrC, int *csrColIndC ) {
return cusparseScsrgemm(handle,
transA, transB,
m, n, k,
descrA, nnzA,
csrValA,
csrRowPtrA, csrColIndA,
descrB, nnzB,
csrValB,
csrRowPtrB, csrColIndB,
descrC,
csrValC,
csrRowPtrC, csrColIndC );
}
void cuSPARSE_apply(
KernelHandle *handle,
typename KernelHandle::row_lno_t m,
typename KernelHandle::row_lno_t n,
typename KernelHandle::row_lno_t k,
in_row_index_view_type row_mapA,
in_nonzero_index_view_type entriesA,
in_nonzero_value_view_type valuesA,
bool transposeA,
in_row_index_view_type row_mapB,
in_nonzero_index_view_type entriesB,
in_nonzero_value_view_type valuesB,
bool transposeB,
typename in_row_index_view_type::non_const_type &row_mapC,
typename in_nonzero_index_view_type::non_const_type &entriesC,
typename in_nonzero_value_view_type::non_const_type &valuesC){
#ifdef KERNELS_HAVE_CUSPARSE
typedef typename KernelHandle::row_lno_t idx;
typedef in_row_index_view_type idx_array_type;
typedef typename KernelHandle::nnz_scalar_t value_type;
typedef typename in_row_index_view_type::device_type device1;
typedef typename in_nonzero_index_view_type::device_type device2;
typedef typename in_nonzero_value_view_type::device_type device3;
std::cout << "RUNNING CUSParse" << std::endl;
if (Kokkos::Impl::is_same<Kokkos::Cuda, device1 >::value){
std::cerr << "MEMORY IS NOT ALLOCATED IN GPU DEVICE for CUSPARSE" << std::endl;
return;
}
if (Kokkos::Impl::is_same<Kokkos::Cuda, device2 >::value){
std::cerr << "MEMORY IS NOT ALLOCATED IN GPU DEVICE for CUSPARSE" << std::endl;
return;
}
if (Kokkos::Impl::is_same<Kokkos::Cuda, device3 >::value){
std::cerr << "MEMORY IS NOT ALLOCATED IN GPU DEVICE for CUSPARSE" << std::endl;
return;
}
if (Kokkos::Impl::is_same<idx, int>::value){
int *a_xadj = (int *)row_mapA.ptr_on_device();
int *b_xadj = (int *)row_mapB.ptr_on_device();
int *c_xadj = (int *)row_mapC.ptr_on_device();
int *a_adj = (int *)entriesA.ptr_on_device();
int *b_adj = (int *)entriesB.ptr_on_device();
int *c_adj = (int *)entriesC.ptr_on_device();
typename KernelHandle::SPGEMMcuSparseHandleType *h = handle->get_cuSparseHandle();
int nnzA = entriesA.dimension_0();
int nnzB = entriesB.dimension_0();
value_type *a_ew = valuesA.ptr_on_device();
value_type *b_ew = valuesB.ptr_on_device();
value_type *c_ew = valuesC.ptr_on_device();
if (Kokkos::Impl::is_same<value_type, float>::value){
cusparseScsrgemm(
h->handle,
h->transA,
h->transB,
m,
n,
k,
h->a_descr,
nnzA,
(float *)a_ew,
a_xadj,
a_adj,
h->b_descr,
nnzB,
(float *)b_ew,
b_xadj,
b_adj,
h->c_descr,
(float *)c_ew,
c_xadj,
c_adj);
}
else if (Kokkos::Impl::is_same<value_type, double>::value){
cusparseDcsrgemm(
h->handle,
h->transA,
h->transB,
m,
n,
k,
h->a_descr,
nnzA,
(double *)a_ew,
a_xadj,
a_adj,
h->b_descr,
nnzB,
(double *)b_ew,
b_xadj,
b_adj,
h->c_descr,
(double *)c_ew,
c_xadj,
c_adj);
}
else {
std::cerr << "CUSPARSE requires float or double values. cuComplex and cuDoubleComplex are not implemented yet." << std::endl;
return;
}
}
else {
std::cerr << "CUSPARSE requires integer values" << std::endl;
return;
}
#else
std::cerr << "CUSPARSE IS NOT DEFINED" << std::endl;
return;
#endif
}
