"tests/python/vscode:/vscode.git/clone" did not exist on "1320438392ae0cb233cc588477ec65d63d86a692"
Unverified Commit 6380ee35 authored by Jeff Rasley's avatar Jeff Rasley Committed by GitHub
Browse files

Fixes for RTD build errors (#606)


Co-authored-by: default avatarShaden Smith <Shaden.Smith@microsoft.com>
parent 007466e5
...@@ -2,8 +2,12 @@ try: ...@@ -2,8 +2,12 @@ try:
# This is populated by setup.py # This is populated by setup.py
from .git_version_info_installed import * from .git_version_info_installed import *
except ModuleNotFoundError: except ModuleNotFoundError:
import os
if os.path.isfile('version.txt'):
# Will be missing from checkouts that haven't been installed (e.g., readthedocs) # Will be missing from checkouts that haven't been installed (e.g., readthedocs)
version = open('version.txt', 'r').read().strip() version = open('version.txt', 'r').read().strip()
else:
version = "0.0.0"
git_hash = '[none]' git_hash = '[none]'
git_branch = '[none]' git_branch = '[none]'
......
...@@ -224,8 +224,8 @@ class Softmax: ...@@ -224,8 +224,8 @@ class Softmax:
For more details about sparsity config, please see `Generative Modeling with Sparse Transformers`: https://arxiv.org/abs/1904.10509 For more details about sparsity config, please see `Generative Modeling with Sparse Transformers`: https://arxiv.org/abs/1904.10509
""" """
def sparse_softmax(*args, **kwargs):
sparse_softmax = _sparse_softmax.apply return _sparse_softmax.apply(*args, **kwargs)
def make_lut(self, device): def make_lut(self, device):
"""Generates the sparsity layout used in block-sparse softmax """Generates the sparsity layout used in block-sparse softmax
......
...@@ -79,4 +79,4 @@ add_module_names = True ...@@ -79,4 +79,4 @@ add_module_names = True
autoclass_content = 'both' autoclass_content = 'both'
autodoc_mock_imports = ["torch", "apex", "mpi4py", "tensorboardX", "numpy"] autodoc_mock_imports = ["torch", "apex", "mpi4py", "tensorboardX", "numpy", "cupy"]
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