Skip to content

Commit 0a10b61

Browse files
author
Svetlana Karslioglu
authored
Merge branch 'master' into tmm
2 parents 3e6a2ce + 72716e6 commit 0a10b61

File tree

3 files changed

+15
-4
lines changed

3 files changed

+15
-4
lines changed

intermediate_source/nvfuser_intro_tutorial.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -557,9 +557,16 @@ def primitive_definition_for_memory_efficient_fusion(
557557
#
558558
# .. figure:: /_static/img/nvfuser_intro/nvfuser_tutorial_5.png
559559
#
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+
563570

564571
######################################################################
565572
# Transformer Block With a Novel Normalization

recipes_source/recipes/saving_multiple_models_in_one_file.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@
5454

5555

5656
######################################################################
57-
# 2. Define and intialize the neural network
57+
# 2. Define and initialize the neural network
5858
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
5959
#
6060
# For sake of example, we will create a neural network for training

requirements.txt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,10 @@ transformers
3131
torchmultimodal-nightly
3232
deep_phonemizer==0.0.17
3333

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+
3438
# PyTorch Theme
3539
-e git+https://github.com/pytorch/pytorch_sphinx_theme.git#egg=pytorch_sphinx_theme
3640

0 commit comments

Comments
 (0)