Unverified Commit c5119480 authored by Hongzhi (Steve), Chen's avatar Hongzhi (Steve), Chen Committed by GitHub
Browse files

[Sparse] Remove unnecessary API exposure. (#5135)



* remove unnecessary exposure.

* remove_add_power

* remove_warning

* reorder
Co-authored-by: default avatarSteve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
parent c6228c01
...@@ -5,9 +5,6 @@ dgl.sparse ...@@ -5,9 +5,6 @@ dgl.sparse
`dgl.sparse` is a library for sparse operators that are commonly used in GNN models. `dgl.sparse` is a library for sparse operators that are commonly used in GNN models.
.. warning::
This is an experimental package. The sparse operators provided in this library do not guarantee the same performance as their message-passing api counterparts.
Sparse matrix class Sparse matrix class
------------------------- -------------------------
.. currentmodule:: dgl.sparse .. currentmodule:: dgl.sparse
...@@ -85,16 +82,15 @@ Attributes and methods ...@@ -85,16 +82,15 @@ Attributes and methods
SparseMatrix.dtype SparseMatrix.dtype
SparseMatrix.device SparseMatrix.device
SparseMatrix.val SparseMatrix.val
SparseMatrix.__repr__
SparseMatrix.row SparseMatrix.row
SparseMatrix.col SparseMatrix.col
SparseMatrix.indices
SparseMatrix.coo SparseMatrix.coo
SparseMatrix.csr SparseMatrix.csr
SparseMatrix.csc SparseMatrix.csc
SparseMatrix.coalesce SparseMatrix.coalesce
SparseMatrix.has_duplicate SparseMatrix.has_duplicate
SparseMatrix.dense SparseMatrix.dense
SparseMatrix.transpose
SparseMatrix.t SparseMatrix.t
SparseMatrix.T SparseMatrix.T
SparseMatrix.to SparseMatrix.to
...@@ -104,7 +100,6 @@ Attributes and methods ...@@ -104,7 +100,6 @@ Attributes and methods
SparseMatrix.double SparseMatrix.double
SparseMatrix.int SparseMatrix.int
SparseMatrix.long SparseMatrix.long
SparseMatrix.transpose
SparseMatrix.reduce SparseMatrix.reduce
SparseMatrix.sum SparseMatrix.sum
SparseMatrix.smax SparseMatrix.smax
...@@ -112,7 +107,6 @@ Attributes and methods ...@@ -112,7 +107,6 @@ Attributes and methods
SparseMatrix.smean SparseMatrix.smean
SparseMatrix.neg SparseMatrix.neg
SparseMatrix.softmax SparseMatrix.softmax
SparseMatrix.__matmul__
Diagonal matrix class Diagonal matrix class
------------------------- -------------------------
...@@ -140,12 +134,11 @@ Attributes and methods ...@@ -140,12 +134,11 @@ Attributes and methods
DiagMatrix.dtype DiagMatrix.dtype
DiagMatrix.device DiagMatrix.device
DiagMatrix.val DiagMatrix.val
DiagMatrix.__repr__
DiagMatrix.as_sparse DiagMatrix.as_sparse
DiagMatrix.dense DiagMatrix.dense
DiagMatrix.transpose
DiagMatrix.t DiagMatrix.t
DiagMatrix.T DiagMatrix.T
DiagMatrix.transpose
DiagMatrix.to DiagMatrix.to
DiagMatrix.cuda DiagMatrix.cuda
DiagMatrix.cpu DiagMatrix.cpu
...@@ -155,7 +148,6 @@ Attributes and methods ...@@ -155,7 +148,6 @@ Attributes and methods
DiagMatrix.long DiagMatrix.long
DiagMatrix.neg DiagMatrix.neg
DiagMatrix.inv DiagMatrix.inv
DiagMatrix.__matmul__
Operators Operators
--------- ---------
...@@ -164,16 +156,6 @@ Operators ...@@ -164,16 +156,6 @@ Operators
.. autosummary:: .. autosummary::
:toctree: ../../generated/ :toctree: ../../generated/
sp_add
sp_mul
sp_power
diag_add
diag_sub
diag_mul
diag_div
diag_power
add
power
spmm spmm
bspmm bspmm
spspmm spspmm
......
...@@ -4,8 +4,6 @@ from typing import Union ...@@ -4,8 +4,6 @@ from typing import Union
from .diag_matrix import diag, DiagMatrix from .diag_matrix import diag, DiagMatrix
from .sparse_matrix import SparseMatrix from .sparse_matrix import SparseMatrix
__all__ = ["diag_add", "diag_sub", "diag_mul", "diag_div", "diag_power"]
def diag_add( def diag_add(
D1: DiagMatrix, D2: Union[DiagMatrix, SparseMatrix] D1: DiagMatrix, D2: Union[DiagMatrix, SparseMatrix]
......
...@@ -6,8 +6,6 @@ import torch ...@@ -6,8 +6,6 @@ import torch
from .diag_matrix import DiagMatrix from .diag_matrix import DiagMatrix
from .sparse_matrix import SparseMatrix, val_like from .sparse_matrix import SparseMatrix, val_like
__all__ = ["sp_add", "sp_mul", "sp_power"]
def spsp_add(A, B): def spsp_add(A, B):
"""Invoke C++ sparse library for addition""" """Invoke C++ sparse library for addition"""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment