Commit 47745860 authored by rusty1s's avatar rusty1s
Browse files

fix doc build

parent 8ca6d861
[run] [run]
source=torch_scatter source=torch_scatter
omit=torch_scatter/placeholder.py
[report] [report]
exclude_lines = exclude_lines =
pragma: no cover pragma: no cover
......
# flake8: noqa
import os import os
import importlib import importlib
import os.path as osp import os.path as osp
...@@ -10,8 +8,9 @@ __version__ = '2.0.3' ...@@ -10,8 +8,9 @@ __version__ = '2.0.3'
expected_torch_version = (1, 4) expected_torch_version = (1, 4)
try: try:
torch.ops.load_library(importlib.machinery.PathFinder().find_spec( for library in ['_version', '_scatter', '_segment_csr', '_segment_coo']:
'_version', [osp.dirname(__file__)]).origin) torch.ops.load_library(importlib.machinery.PathFinder().find_spec(
library, [osp.dirname(__file__)]).origin)
except OSError as e: except OSError as e:
if 'undefined symbol' in str(e): if 'undefined symbol' in str(e):
major, minor = [int(x) for x in torch.__version__.split('.')[:2]] major, minor = [int(x) for x in torch.__version__.split('.')[:2]]
...@@ -25,6 +24,31 @@ except AttributeError as e: ...@@ -25,6 +24,31 @@ except AttributeError as e:
if os.getenv('BUILD_DOCS', '0') != '1': if os.getenv('BUILD_DOCS', '0') != '1':
raise AttributeError(e) raise AttributeError(e)
from .placeholder import cuda_version_placeholder
torch.ops.torch_scatter.cuda_version = cuda_version_placeholder
from .placeholder import scatter_arg_placeholder
torch.ops.torch_scatter.scatter_min = scatter_arg_placeholder
torch.ops.torch_scatter.scatter_max = scatter_arg_placeholder
from .placeholder import segment_csr_placeholder
from .placeholder import segment_csr_arg_placeholder
from .placeholder import gather_csr_placeholder
torch.ops.torch_scatter.segment_sum_csr = segment_csr_placeholder
torch.ops.torch_scatter.segment_mean_csr = segment_csr_placeholder
torch.ops.torch_scatter.segment_min_csr = segment_csr_arg_placeholder
torch.ops.torch_scatter.segment_max_csr = segment_csr_arg_placeholder
torch.ops.torch_scatter.gather_csr = gather_csr_placeholder
from .placeholder import segment_coo_placeholder
from .placeholder import segment_coo_arg_placeholder
from .placeholder import gather_coo_placeholder
torch.ops.torch_scatter.segment_sum_coo = segment_coo_placeholder
torch.ops.torch_scatter.segment_mean_coo = segment_coo_placeholder
torch.ops.torch_scatter.segment_min_coo = segment_coo_arg_placeholder
torch.ops.torch_scatter.segment_max_coo = segment_coo_arg_placeholder
torch.ops.torch_scatter.gather_coo = gather_coo_placeholder
if torch.version.cuda is not None: # pragma: no cover if torch.version.cuda is not None: # pragma: no cover
cuda_version = torch.ops.torch_scatter.cuda_version() cuda_version = torch.ops.torch_scatter.cuda_version()
...@@ -45,15 +69,15 @@ if torch.version.cuda is not None: # pragma: no cover ...@@ -45,15 +69,15 @@ if torch.version.cuda is not None: # pragma: no cover
f'matches your PyTorch install.') f'matches your PyTorch install.')
from .scatter import (scatter_sum, scatter_add, scatter_mean, scatter_min, from .scatter import (scatter_sum, scatter_add, scatter_mean, scatter_min,
scatter_max, scatter) scatter_max, scatter) # noqa: E402
from .segment_csr import (segment_sum_csr, segment_add_csr, segment_mean_csr, from .segment_csr import (segment_sum_csr, segment_add_csr, segment_mean_csr,
segment_min_csr, segment_max_csr, segment_csr, segment_min_csr, segment_max_csr, segment_csr,
gather_csr) gather_csr) # noqa: E402
from .segment_coo import (segment_sum_coo, segment_add_coo, segment_mean_coo, from .segment_coo import (segment_sum_coo, segment_add_coo, segment_mean_coo,
segment_min_coo, segment_max_coo, segment_coo, segment_min_coo, segment_max_coo, segment_coo,
gather_coo) gather_coo) # noqa: E402
from .composite import (scatter_std, scatter_logsumexp, scatter_softmax, from .composite import (scatter_std, scatter_logsumexp, scatter_softmax,
scatter_log_softmax) scatter_log_softmax) # noqa: E402
__all__ = [ __all__ = [
'scatter_sum', 'scatter_sum',
......
from typing import Optional, Tuple
import torch
def cuda_version_placeholder() -> int:
return -1
def scatter_placeholder(src: torch.Tensor, index: torch.Tensor, dim: int,
out: Optional[torch.Tensor],
dim_size: Optional[int]) -> torch.Tensor:
raise ImportError
return src
def scatter_arg_placeholder(src: torch.Tensor, index: torch.Tensor, dim: int,
out: Optional[torch.Tensor],
dim_size: Optional[int]
) -> Tuple[torch.Tensor, torch.Tensor]:
raise ImportError
return src, index
def segment_csr_placeholder(src: torch.Tensor, indptr: torch.Tensor,
out: Optional[torch.Tensor]) -> torch.Tensor:
raise ImportError
return src
def segment_csr_arg_placeholder(src: torch.Tensor, indptr: torch.Tensor,
out: Optional[torch.Tensor]
) -> Tuple[torch.Tensor, torch.Tensor]:
raise ImportError
return src, indptr
def gather_csr_placeholder(src: torch.Tensor, indptr: torch.Tensor,
out: Optional[torch.Tensor]) -> torch.Tensor:
raise ImportError
return src
def segment_coo_placeholder(src: torch.Tensor, index: torch.Tensor,
out: Optional[torch.Tensor],
dim_size: Optional[int]) -> torch.Tensor:
raise ImportError
return src
def segment_coo_arg_placeholder(src: torch.Tensor, index: torch.Tensor,
out: Optional[torch.Tensor],
dim_size: Optional[int]
) -> Tuple[torch.Tensor, torch.Tensor]:
raise ImportError
return src, index
def gather_coo_placeholder(src: torch.Tensor, index: torch.Tensor,
out: Optional[torch.Tensor]) -> torch.Tensor:
raise ImportError
return src
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