Add nnz argument to set_csr_data in sparse CG examples

Author: noffermans

Libraries/oneMKL/sparse_conjugate_gradient/sparse_cg.cpp

@@ -380,7 +380,7 @@ int run_sparse_pcg_example(const sycl::device &dev)
         // setup optimizations and properties we know about A matrix
         oneapi::mkl::sparse::init_matrix_handle(&A);
 
-        auto ev_set = oneapi::mkl::sparse::set_csr_data(q, A, n, n,
+        auto ev_set = oneapi::mkl::sparse::set_csr_data(q, A, n, n, nnz,
                 oneapi::mkl::index_base::zero, ia_d, ja_d, a_d, {});
 
         oneapi::mkl::sparse::set_matrix_property(A, oneapi::mkl::sparse::property::symmetric);

Libraries/oneMKL/sparse_conjugate_gradient/sparse_cg2.cpp

@@ -453,7 +453,7 @@ int run_sparse_pcg_example(const sycl::device &dev)
         // setup optimizations and properties we know about A matrix
         oneapi::mkl::sparse::init_matrix_handle(&A);
 
-        auto ev_set = oneapi::mkl::sparse::set_csr_data(q, A, n, n,
+        auto ev_set = oneapi::mkl::sparse::set_csr_data(q, A, n, n, nnz,
                 oneapi::mkl::index_base::zero, ia_d, ja_d, a_d, {});
 
         oneapi::mkl::sparse::set_matrix_property(A, oneapi::mkl::sparse::property::symmetric);