File tree 1 file changed +10
-3
lines changed
1 file changed +10
-3
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
You can’t perform that action at this time.
0 commit comments