Skip to content

[1.13] Update nvfuser tutorial #2098

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Oct 24, 2022
Merged
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
13 changes: 10 additions & 3 deletions intermediate_source/nvfuser_intro_tutorial.py
Original file line number Diff line number Diff line change
Expand Up @@ -557,9 +557,16 @@ def primitive_definition_for_memory_efficient_fusion(
#
# .. figure:: /_static/img/nvfuser_intro/nvfuser_tutorial_5.png
#
# .. note:: FuncTorch’s memory efficient pass is still actively in development
# and future versions are expected to achieve performance closer
# to that of TorchScript with the composite definition.
# .. note:: FuncTorch’s memory efficient pass is experimental and still
# actively in development.
# Future versions of the API are expected to achieve performance
# closer to that of TorchScript with the composite definition.
#
# .. note:: FuncTorch’s memory efficient pass specializes on the shapes of
# the inputs to the function. If new inputs are provided with
# different shapes, then you need to construct a new function
# using `memory_efficient_fusion` and apply it to the new inputs.


######################################################################
# Transformer Block With a Novel Normalization
Expand Down