File tree Expand file tree Collapse file tree 3 files changed +15
-4
lines changed Expand file tree Collapse file tree 3 files changed +15
-4
lines changed Original file line number Diff line number Diff line change @@ -557,9 +557,16 @@ def primitive_definition_for_memory_efficient_fusion(
557
557
#
558
558
# .. figure:: /_static/img/nvfuser_intro/nvfuser_tutorial_5.png
559
559
#
560
- # .. note:: FuncTorch’s memory efficient pass is still actively in development
561
- # and future versions are expected to achieve performance closer
562
- # to that of TorchScript with the composite definition.
560
+ # .. note:: FuncTorch’s memory efficient pass is experimental and still
561
+ # actively in development.
562
+ # Future versions of the API are expected to achieve performance
563
+ # closer to that of TorchScript with the composite definition.
564
+ #
565
+ # .. note:: FuncTorch’s memory efficient pass specializes on the shapes of
566
+ # the inputs to the function. If new inputs are provided with
567
+ # different shapes, then you need to construct a new function
568
+ # using `memory_efficient_fusion` and apply it to the new inputs.
569
+
563
570
564
571
######################################################################
565
572
# Transformer Block With a Novel Normalization
Original file line number Diff line number Diff line change 54
54
55
55
56
56
######################################################################
57
- # 2. Define and intialize the neural network
57
+ # 2. Define and initialize the neural network
58
58
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
59
59
#
60
60
# For sake of example, we will create a neural network for training
Original file line number Diff line number Diff line change @@ -31,6 +31,10 @@ transformers
31
31
torchmultimodal-nightly
32
32
deep_phonemizer == 0.0.17
33
33
34
+ # the following is necessary due to https://github.com/python/importlib_metadata/issues/411
35
+ importlib-metadata < 5.0 ; python_version < = "3.7"
36
+ importlib-metadata ; python_version > "3.7"
37
+
34
38
# PyTorch Theme
35
39
-e git+https://github.com/pytorch/pytorch_sphinx_theme.git#egg=pytorch_sphinx_theme
36
40
You can’t perform that action at this time.
0 commit comments