chore: import upstream snapshot with attribution

2026-07-13 13:35:51 +08:00
commit c36a561cd8
2172 changed files with 455595 additions and 0 deletions
@@ -0,0 +1,53 @@
+/**
+ *  Copyright (c) 2022 by Contributors
+ * @file sparse/elementwise_op.h
+ * @brief DGL C++ sparse elementwise operators.
+ */
+#ifndef SPARSE_ELEMENTWISE_OP_H_
+#define SPARSE_ELEMENTWISE_OP_H_
+
+#include <sparse/sparse_matrix.h>
+
+namespace dgl {
+namespace sparse {
+
+/**
+ * @brief Adds two sparse matrices possibly with different sparsities.
+ *
+ * @param lhs_mat SparseMatrix
+ * @param rhs_mat SparseMatrix
+ *
+ * @return SparseMatrix
+ */
+c10::intrusive_ptr<SparseMatrix> SpSpAdd(
+    const c10::intrusive_ptr<SparseMatrix>& lhs_mat,
+    const c10::intrusive_ptr<SparseMatrix>& rhs_mat);
+
+/**
+ * @brief Multiplies two sparse matrices possibly with different sparsities.
+ *
+ * @param lhs_mat SparseMatrix
+ * @param rhs_mat SparseMatrix
+ *
+ * @return SparseMatrix
+ */
+c10::intrusive_ptr<SparseMatrix> SpSpMul(
+    const c10::intrusive_ptr<SparseMatrix>& lhs_mat,
+    const c10::intrusive_ptr<SparseMatrix>& rhs_mat);
+
+/**
+ * @brief Divides two sparse matrices with the same sparsity.
+ *
+ * @param lhs_mat SparseMatrix
+ * @param rhs_mat SparseMatrix
+ *
+ * @return SparseMatrix
+ */
+c10::intrusive_ptr<SparseMatrix> SpSpDiv(
+    const c10::intrusive_ptr<SparseMatrix>& lhs_mat,
+    const c10::intrusive_ptr<SparseMatrix>& rhs_mat);
+
+}  // namespace sparse
+}  // namespace dgl
+
+#endif  // SPARSE_ELEMENTWISE_OP_H_