Last changes

pytorch · Dec 23, 2024 · 0597d3a · 0597d3a
1 parent e297c9b
commit 0597d3a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/exir/emit/_emitter.py b/exir/emit/_emitter.py
@@ -1575,7 +1575,8 @@ def _find_fqn_for_placeholder(
                     warnings.warn(
                         "Mutation on a buffer in the model is detected. ExecuTorch assumes "
                         "buffers that are mutated in the graph have a meaningless initial state, "
-                        "only the shape and dtype will be serialized.",
+                        "only the shape and dtype will be serialized, unless a pass which marks "
+                        "spec.const=True such as InitializedMutableBufferPass is run.",
                         UserWarning,
                         stacklevel=1,
                     )

diff --git a/exir/passes/init_mutable_pass.py b/exir/passes/init_mutable_pass.py
@@ -26,6 +26,7 @@ def __init__(self, patterns: List[str]) -> None:
     def placeholder(self, name: str, arg, meta):
         for pattern in self.patterns:
             if pattern in name:
+                breakpoint()
                 meta["et_init_buffer"] = True
 
         return super().placeholder(name, arg, meta)