25561548952

Committed 08 May 2026 02:36PM UTC coverage: 77.679% (-0.8%) from 78.441%

Build # 25561548952

Build Type

Pull #2841

github

Committed by

web-flow

Commit Message

Merge f38ae7b4b into a3cced712

Pull Request Pull Request #2841: Feature sparse linalg solvers

Coverage Stats

1573 of 2960 branches covered (53.14%)

Branch coverage included in aggregate %.

508 of 951 new or added lines in 7 files covered. (53.42%)

23 existing lines in 1 file now uncovered.

26779 of 33539 relevant lines covered (79.84%)

7900.22 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

0.0

/dpnp/backend/extensions/sparse/sparse_py.cpp

//*****************************************************************************
// Copyright (c) 2025, Intel Corporation
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
// - Redistributions of source code must retain the above copyright notice,
//   this list of conditions and the following disclaimer.
// - Redistributions in binary form must reproduce the above copyright notice,
//   this list of conditions and the following disclaimer in the documentation
//   and/or other materials provided with the distribution.
// - Neither the name of the copyright holder nor the names of its contributors
//   may be used to endorse or promote products derived from this software
//   without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.
//*****************************************************************************

#include <pybind11/pybind11.h>
#include <pybind11/stl.h>

#include <cstdint>
#include <tuple>
#include <vector>

#include "gemv.hpp"

namespace py = pybind11;

using dpnp::extensions::sparse::init_sparse_gemv_dispatch_tables;
using dpnp::extensions::sparse::sparse_gemv_compute;
using dpnp::extensions::sparse::sparse_gemv_init;
using dpnp::extensions::sparse::sparse_gemv_release;

PYBIND11_MODULE(_sparse_impl, m)
{
    init_sparse_gemv_dispatch_tables();

    // ------------------------------------------------------------------
    // _using_onemath()
    //
    // Reports whether the module was compiled against the portable
    // OneMath interface (USE_ONEMATH) rather than direct oneMKL.
    // ------------------------------------------------------------------
    m.def("_using_onemath", []() -> bool {
#ifdef USE_ONEMATH
        return true;
#else
        return false;
#endif
    });

    // ------------------------------------------------------------------
    // _sparse_gemv_init(exec_q, trans, row_ptr, col_ind, values,
    //                   num_rows, num_cols, nnz, depends)
    //     -> (handle: int, val_type_id: int, event)
    //
    // Calls init_matrix_handle + set_csr_data + optimize_gemv ONCE.
    //
    // The returned handle is an opaque uintptr_t; val_type_id is the
    // dpctl typenum lookup id of the matrix value dtype and MUST be
    // passed back to _sparse_gemv_compute so the C++ layer can verify
    // that x and y dtype match the handle.
    //
    // LIFETIME CONTRACT: the caller must keep row_ptr / col_ind / values
    // USM allocations alive until _sparse_gemv_release has been called
    // AND its returned event has completed. The handle does not copy
    // the CSR arrays.
    // ------------------------------------------------------------------
    m.def(
        "_sparse_gemv_init",
        [](sycl::queue &exec_q, const int trans,
           const dpnp::tensor::usm_ndarray &row_ptr,
           const dpnp::tensor::usm_ndarray &col_ind,
           const dpnp::tensor::usm_ndarray &values, const std::int64_t num_rows,
           const std::int64_t num_cols, const std::int64_t nnz,
           const std::vector<sycl::event> &depends)
            -> std::tuple<std::uintptr_t, int, sycl::event> {
            return sparse_gemv_init(exec_q, trans, row_ptr, col_ind, values,
                                    num_rows, num_cols, nnz, depends);
        },
        py::arg("exec_q"), py::arg("trans"), py::arg("row_ptr"),
        py::arg("col_ind"), py::arg("values"), py::arg("num_rows"),
        py::arg("num_cols"), py::arg("nnz"), py::arg("depends"),
        "Initialise oneMKL sparse matrix handle "
        "(set_csr_data + optimize_gemv). "
        "Returns (handle_ptr: int, val_type_id: int, event). "
        "Call once per operator.");

    // ------------------------------------------------------------------
    // _sparse_gemv_compute(exec_q, handle, val_type_id, trans, alpha,
    //                     x, beta, y, num_rows, num_cols, depends)
    //     -> gemv_event
    //
    // Fires sparse::gemv using a pre-built handle. Verifies x and y
    // dtype match val_type_id from init, and that shapes agree with
    // op(A) dimensions (swapped for trans != N).
    //
    // Only the cheap MKL kernel is dispatched; no analysis overhead.
    // No host_task keep-alive is submitted -- pybind11 refcounts the
    // usm_ndarrays across the call, and sequencing of subsequent work
    // on the same queue happens automatically.
    // ------------------------------------------------------------------
    m.def(
        "_sparse_gemv_compute",
        [](sycl::queue &exec_q, const std::uintptr_t handle_ptr,
           const int val_type_id, const int trans, const double alpha,
           const dpnp::tensor::usm_ndarray &x, const double beta,
           const dpnp::tensor::usm_ndarray &y, const std::int64_t num_rows,
           const std::int64_t num_cols,
           const std::vector<sycl::event> &depends) -> sycl::event {
            return sparse_gemv_compute(exec_q, handle_ptr, val_type_id, trans,
                                       alpha, x, beta, y, num_rows, num_cols,
                                       depends);
        },
        py::arg("exec_q"), py::arg("handle"), py::arg("val_type_id"),
        py::arg("trans"), py::arg("alpha"), py::arg("x"), py::arg("beta"),
        py::arg("y"), py::arg("num_rows"), py::arg("num_cols"),
        py::arg("depends"),
        "Execute sparse::gemv using a pre-built handle. "
        "Returns the gemv event.");

    // ------------------------------------------------------------------
    // _sparse_gemv_release(exec_q, handle, depends) -> event
    //
    // Releases the matrix_handle allocated by _sparse_gemv_init.
    // Must be called exactly once per handle after all compute calls
    // referencing it have completed. The returned event depends on the
    // release, so callers can chain CSR buffer deallocation on it.
    // ------------------------------------------------------------------
    m.def(
        "_sparse_gemv_release",
        [](sycl::queue &exec_q, const std::uintptr_t handle_ptr,
           const std::vector<sycl::event> &depends) -> sycl::event {
            return sparse_gemv_release(exec_q, handle_ptr, depends);
        },
        py::arg("exec_q"), py::arg("handle"), py::arg("depends"),
        "Release the oneMKL matrix_handle created by _sparse_gemv_init.");
}

1	//*****************************************************************************
2	// Copyright (c) 2025, Intel Corporation
3	// All rights reserved.
4	//
5	// Redistribution and use in source and binary forms, with or without
6	// modification, are permitted provided that the following conditions are met:
7	// - Redistributions of source code must retain the above copyright notice,
8	// this list of conditions and the following disclaimer.
9	// - Redistributions in binary form must reproduce the above copyright notice,
10	// this list of conditions and the following disclaimer in the documentation
11	// and/or other materials provided with the distribution.
12	// - Neither the name of the copyright holder nor the names of its contributors
13	// may be used to endorse or promote products derived from this software
14	// without specific prior written permission.
15	//
16	// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
17	// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18	// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19	// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
20	// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21	// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22	// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23	// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24	// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25	// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26	// POSSIBILITY OF SUCH DAMAGE.
27	//*****************************************************************************
28
29	#include <pybind11/pybind11.h>
30	#include <pybind11/stl.h>
31
32	#include <cstdint>
33	#include <tuple>
34	#include <vector>
35
36	#include "gemv.hpp"
37
38	namespace py = pybind11;
39
40	using dpnp::extensions::sparse::init_sparse_gemv_dispatch_tables;
41	using dpnp::extensions::sparse::sparse_gemv_compute;
42	using dpnp::extensions::sparse::sparse_gemv_init;
43	using dpnp::extensions::sparse::sparse_gemv_release;
44
45	PYBIND11_MODULE(_sparse_impl, m)
NEW 46	{	×
NEW 47	init_sparse_gemv_dispatch_tables();	×
48
49	// ------------------------------------------------------------------
50	// _using_onemath()
51	//
52	// Reports whether the module was compiled against the portable
53	// OneMath interface (USE_ONEMATH) rather than direct oneMKL.
54	// ------------------------------------------------------------------
NEW 55	m.def("_using_onemath", []() -> bool {	×
56	#ifdef USE_ONEMATH
57	return true;
58	#else
NEW 59	return false;	×
NEW 60	#endif	×
NEW 61	});	×
62
63	// ------------------------------------------------------------------
64	// _sparse_gemv_init(exec_q, trans, row_ptr, col_ind, values,
65	// num_rows, num_cols, nnz, depends)
66	// -> (handle: int, val_type_id: int, event)
67	//
68	// Calls init_matrix_handle + set_csr_data + optimize_gemv ONCE.
69	//
70	// The returned handle is an opaque uintptr_t; val_type_id is the
71	// dpctl typenum lookup id of the matrix value dtype and MUST be
72	// passed back to _sparse_gemv_compute so the C++ layer can verify
73	// that x and y dtype match the handle.
74	//
75	// LIFETIME CONTRACT: the caller must keep row_ptr / col_ind / values
76	// USM allocations alive until _sparse_gemv_release has been called
77	// AND its returned event has completed. The handle does not copy
78	// the CSR arrays.
79	// ------------------------------------------------------------------
NEW 80	m.def(	×
NEW 81	"_sparse_gemv_init",	×
NEW 82	[](sycl::queue &exec_q, const int trans,	×
NEW 83	const dpnp::tensor::usm_ndarray &row_ptr,	×
NEW 84	const dpnp::tensor::usm_ndarray &col_ind,	×
NEW 85	const dpnp::tensor::usm_ndarray &values, const std::int64_t num_rows,	×
NEW 86	const std::int64_t num_cols, const std::int64_t nnz,	×
NEW 87	const std::vector<sycl::event> &depends)	×
NEW 88	-> std::tuple<std::uintptr_t, int, sycl::event> {	×
NEW 89	return sparse_gemv_init(exec_q, trans, row_ptr, col_ind, values,	×
NEW 90	num_rows, num_cols, nnz, depends);	×
NEW 91	},	×
NEW 92	py::arg("exec_q"), py::arg("trans"), py::arg("row_ptr"),	×
NEW 93	py::arg("col_ind"), py::arg("values"), py::arg("num_rows"),	×
NEW 94	py::arg("num_cols"), py::arg("nnz"), py::arg("depends"),	×
NEW 95	"Initialise oneMKL sparse matrix handle "	×
NEW 96	"(set_csr_data + optimize_gemv). "	×
NEW 97	"Returns (handle_ptr: int, val_type_id: int, event). "	×
NEW 98	"Call once per operator.");	×
99
100	// ------------------------------------------------------------------
101	// _sparse_gemv_compute(exec_q, handle, val_type_id, trans, alpha,
102	// x, beta, y, num_rows, num_cols, depends)
103	// -> gemv_event
104	//
105	// Fires sparse::gemv using a pre-built handle. Verifies x and y
106	// dtype match val_type_id from init, and that shapes agree with
107	// op(A) dimensions (swapped for trans != N).
108	//
109	// Only the cheap MKL kernel is dispatched; no analysis overhead.
110	// No host_task keep-alive is submitted -- pybind11 refcounts the
111	// usm_ndarrays across the call, and sequencing of subsequent work
112	// on the same queue happens automatically.
113	// ------------------------------------------------------------------
NEW 114	m.def(	×
NEW 115	"_sparse_gemv_compute",	×
NEW 116	[](sycl::queue &exec_q, const std::uintptr_t handle_ptr,	×
NEW 117	const int val_type_id, const int trans, const double alpha,	×
NEW 118	const dpnp::tensor::usm_ndarray &x, const double beta,	×
NEW 119	const dpnp::tensor::usm_ndarray &y, const std::int64_t num_rows,	×
NEW 120	const std::int64_t num_cols,	×
NEW 121	const std::vector<sycl::event> &depends) -> sycl::event {	×
NEW 122	return sparse_gemv_compute(exec_q, handle_ptr, val_type_id, trans,	×
NEW 123	alpha, x, beta, y, num_rows, num_cols,	×
NEW 124	depends);	×
NEW 125	},	×
NEW 126	py::arg("exec_q"), py::arg("handle"), py::arg("val_type_id"),	×
NEW 127	py::arg("trans"), py::arg("alpha"), py::arg("x"), py::arg("beta"),	×
NEW 128	py::arg("y"), py::arg("num_rows"), py::arg("num_cols"),	×
NEW 129	py::arg("depends"),	×
NEW 130	"Execute sparse::gemv using a pre-built handle. "	×
NEW 131	"Returns the gemv event.");	×
132
133	// ------------------------------------------------------------------
134	// _sparse_gemv_release(exec_q, handle, depends) -> event
135	//
136	// Releases the matrix_handle allocated by _sparse_gemv_init.
137	// Must be called exactly once per handle after all compute calls
138	// referencing it have completed. The returned event depends on the
139	// release, so callers can chain CSR buffer deallocation on it.
140	// ------------------------------------------------------------------
NEW 141	m.def(	×
NEW 142	"_sparse_gemv_release",	×
NEW 143	[](sycl::queue &exec_q, const std::uintptr_t handle_ptr,	×
NEW 144	const std::vector<sycl::event> &depends) -> sycl::event {	×
NEW 145	return sparse_gemv_release(exec_q, handle_ptr, depends);	×
NEW 146	},	×
NEW 147	py::arg("exec_q"), py::arg("handle"), py::arg("depends"),	×
NEW 148	"Release the oneMKL matrix_handle created by _sparse_gemv_init.");	×
NEW 149	}	×

IntelPython / dpnp / 25561548952

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous