Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 6 additions & 2 deletions deepspeed/git_version_info.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,12 @@
# This is populated by setup.py
from .git_version_info_installed import *
except ModuleNotFoundError:
# Will be missing from checkouts that haven't been installed (e.g., readthedocs)
version = open('version.txt', 'r').read().strip()
import os
if os.path.isfile('version.txt'):
# Will be missing from checkouts that haven't been installed (e.g., readthedocs)
version = open('version.txt', 'r').read().strip()
else:
version = "0.0.0"
git_hash = '[none]'
git_branch = '[none]'

Expand Down
4 changes: 2 additions & 2 deletions deepspeed/ops/sparse_attention/softmax.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
"""

sparse_softmax = _sparse_softmax.apply
def sparse_softmax(*args, **kwargs):
return _sparse_softmax.apply(*args, **kwargs)

def make_lut(self, device):
"""Generates the sparsity layout used in block-sparse softmax
Expand Down
2 changes: 1 addition & 1 deletion docs/code-docs/source/conf.py
Original file line number Diff line number Diff line change
Expand Up @@ -79,4 +79,4 @@

autoclass_content = 'both'

autodoc_mock_imports = ["torch", "apex", "mpi4py", "tensorboardX", "numpy"]
autodoc_mock_imports = ["torch", "apex", "mpi4py", "tensorboardX", "numpy", "cupy"]
1 change: 0 additions & 1 deletion requirements/requirements-readthedocs.txt
Original file line number Diff line number Diff line change
@@ -1,2 +1 @@
tqdm
psutil