new bindings

Author: dance858

sparsediffpy/_bindings/atoms/dense_matmul.h

@@ -0,0 +1,101 @@
+#ifndef ATOM_DENSE_MATMUL_H
+#define ATOM_DENSE_MATMUL_H
+
+#include "bivariate.h"
+#include "common.h"
+
+/* Dense left matrix multiplication: A @ f(x) where A is a dense matrix.
+ *
+ * Python signature:
+ *   make_dense_left_matmul(child, A_data_flat, m, n)
+ *
+ * - child: the child expression capsule f(x).
+ * - A_data_flat: contiguous row-major numpy float64 array of size m*n.
+ * - m, n: dimensions of matrix A. */
+static PyObject *py_make_dense_left_matmul(PyObject *self, PyObject *args)
+{
+    PyObject *child_capsule;
+    PyObject *data_obj;
+    int m, n;
+    if (!PyArg_ParseTuple(args, "OOii", &child_capsule, &data_obj, &m, &n))
+    {
+        return NULL;
+    }
+
+    expr *child = (expr *) PyCapsule_GetPointer(child_capsule, EXPR_CAPSULE_NAME);
+    if (!child)
+    {
+        PyErr_SetString(PyExc_ValueError, "invalid child capsule");
+        return NULL;
+    }
+
+    PyArrayObject *data_array =
+        (PyArrayObject *) PyArray_FROM_OTF(data_obj, NPY_DOUBLE, NPY_ARRAY_IN_ARRAY);
+    if (!data_array)
+    {
+        return NULL;
+    }
+
+    double *A_data = (double *) PyArray_DATA(data_array);
+
+    expr *node = new_left_matmul_dense(child, m, n, A_data);
+    Py_DECREF(data_array);
+
+    if (!node)
+    {
+        PyErr_SetString(PyExc_RuntimeError,
+                        "failed to create dense_left_matmul node");
+        return NULL;
+    }
+    expr_retain(node);
+    return PyCapsule_New(node, EXPR_CAPSULE_NAME, expr_capsule_destructor);
+}
+
+/* Dense right matrix multiplication: f(x) @ A where A is a dense matrix.
+ *
+ * Python signature:
+ *   make_dense_right_matmul(child, A_data_flat, m, n)
+ *
+ * - child: the child expression capsule f(x).
+ * - A_data_flat: contiguous row-major numpy float64 array of size m*n.
+ * - m, n: dimensions of matrix A. */
+static PyObject *py_make_dense_right_matmul(PyObject *self, PyObject *args)
+{
+    PyObject *child_capsule;
+    PyObject *data_obj;
+    int m, n;
+    if (!PyArg_ParseTuple(args, "OOii", &child_capsule, &data_obj, &m, &n))
+    {
+        return NULL;
+    }
+
+    expr *child = (expr *) PyCapsule_GetPointer(child_capsule, EXPR_CAPSULE_NAME);
+    if (!child)
+    {
+        PyErr_SetString(PyExc_ValueError, "invalid child capsule");
+        return NULL;
+    }
+
+    PyArrayObject *data_array =
+        (PyArrayObject *) PyArray_FROM_OTF(data_obj, NPY_DOUBLE, NPY_ARRAY_IN_ARRAY);
+    if (!data_array)
+    {
+        return NULL;
+    }
+
+    double *A_data = (double *) PyArray_DATA(data_array);
+
+    expr *node = new_right_matmul_dense(child, m, n, A_data);
+    Py_DECREF(data_array);
+
+    if (!node)
+    {
+        PyErr_SetString(PyExc_RuntimeError,
+                        "failed to create dense_right_matmul node");
+        return NULL;
+    }
+    expr_retain(node);
+    return PyCapsule_New(node, EXPR_CAPSULE_NAME, expr_capsule_destructor);
+}
+
+#endif /* ATOM_DENSE_MATMUL_H */

sparsediffpy/_bindings/atoms/left_matmul.h

@@ -5,7 +5,7 @@
 #include "common.h"
 
 /* Left matrix multiplication: A @ f(x) where A is a constant matrix */
-static PyObject *py_make_left_matmul(PyObject *self, PyObject *args)
+static PyObject *py_make_sparse_left_matmul(PyObject *self, PyObject *args)
 {
     PyObject *child_capsule;
     PyObject *data_obj, *indices_obj, *indptr_obj;

sparsediffpy/_bindings/atoms/right_matmul.h

@@ -5,7 +5,7 @@
 #include "common.h"
 
 /* Right matrix multiplication: f(x) @ A where A is a constant matrix */
-static PyObject *py_make_right_matmul(PyObject *self, PyObject *args)
+static PyObject *py_make_sparse_right_matmul(PyObject *self, PyObject *args)
 {
     PyObject *child_capsule;
     PyObject *data_obj, *indices_obj, *indptr_obj;

sparsediffpy/_bindings/bindings.c

@@ -11,6 +11,7 @@
 #include "atoms/const_vector_mult.h"
 #include "atoms/constant.h"
 #include "atoms/cos.h"
+#include "atoms/dense_matmul.h"
 #include "atoms/diag_vec.h"
 #include "atoms/entr.h"
 #include "atoms/exp.h"
@@ -109,10 +110,14 @@ static PyMethodDef DNLPMethods[] = {
     {"make_entr", py_make_entr, METH_VARARGS, "Create entr node"},
     {"make_logistic", py_make_logistic, METH_VARARGS, "Create logistic node"},
     {"make_xexp", py_make_xexp, METH_VARARGS, "Create xexp node"},
-    {"make_left_matmul", py_make_left_matmul, METH_VARARGS,
-     "Create left matmul node (A @ f(x))"},
-    {"make_right_matmul", py_make_right_matmul, METH_VARARGS,
-     "Create right matmul node (f(x) @ A)"},
+    {"make_sparse_left_matmul", py_make_sparse_left_matmul, METH_VARARGS,
+     "Create sparse left matmul node (A @ f(x))"},
+    {"make_dense_left_matmul", py_make_dense_left_matmul, METH_VARARGS,
+     "Create dense left matmul node (A @ f(x)) where A is dense"},
+    {"make_sparse_right_matmul", py_make_sparse_right_matmul, METH_VARARGS,
+     "Create sparse right matmul node (f(x) @ A)"},
+    {"make_dense_right_matmul", py_make_dense_right_matmul, METH_VARARGS,
+     "Create dense right matmul node (f(x) @ A) where A is dense"},
     {"make_quad_form", py_make_quad_form, METH_VARARGS,
      "Create quadratic form node (x' * Q * x)"},
     {"make_quad_over_lin", py_make_quad_over_lin, METH_VARARGS,