keep input mutations in the joint graph (#32)

bdhirsh · web-flow · commit 233d68b882f9 · 2025-07-17T16:34:38.000-07:00
diff --git a/autoparallel/export_module.py b/autoparallel/export_module.py
@@ -120,6 +120,7 @@ def forward(self, *args, **kwargs):
             no_tangents=False,
             pre_dispatch=pre_dispatch,
             dynamic_shapes=dynamic_shapes,
+            keep_input_mutations=True,
             kwargs=kwargs,
         )
 

Original file line number	Diff line number	Diff line change
`@@ -120,6 +120,7 @@ def forward(self, args, *kwargs):`
`120`	`120`	`no_tangents=False,`
`121`	`121`	`pre_dispatch=pre_dispatch,`
`122`	`122`	`dynamic_shapes=dynamic_shapes,`
	`123`	`+ keep_input_mutations=True,`
`123`	`124`	`kwargs=kwargs,`
`124`	`125`	`)`
`125`	`126`