Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
deepspeed
Commits
6380ee35
Unverified
Commit
6380ee35
authored
Dec 15, 2020
by
Jeff Rasley
Committed by
GitHub
Dec 15, 2020
Browse files
Fixes for RTD build errors (#606)
Co-authored-by:
Shaden Smith
<
Shaden.Smith@microsoft.com
>
parent
007466e5
Changes
4
Show whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
9 additions
and
6 deletions
+9
-6
deepspeed/git_version_info.py
deepspeed/git_version_info.py
+6
-2
deepspeed/ops/sparse_attention/softmax.py
deepspeed/ops/sparse_attention/softmax.py
+2
-2
docs/code-docs/source/conf.py
docs/code-docs/source/conf.py
+1
-1
requirements/requirements-readthedocs.txt
requirements/requirements-readthedocs.txt
+0
-1
No files found.
deepspeed/git_version_info.py
View file @
6380ee35
...
...
@@ -2,8 +2,12 @@ try:
# This is populated by setup.py
from
.git_version_info_installed
import
*
except
ModuleNotFoundError
:
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]'
...
...
deepspeed/ops/sparse_attention/softmax.py
View file @
6380ee35
...
...
@@ -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
...
...
docs/code-docs/source/conf.py
View file @
6380ee35
...
...
@@ -79,4 +79,4 @@ add_module_names = True
autoclass_content
=
'both'
autodoc_mock_imports
=
[
"torch"
,
"apex"
,
"mpi4py"
,
"tensorboardX"
,
"numpy"
]
autodoc_mock_imports
=
[
"torch"
,
"apex"
,
"mpi4py"
,
"tensorboardX"
,
"numpy"
,
"cupy"
]
requirements/requirements-readthedocs.txt
View file @
6380ee35
tqdm
psutil
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment