Skip to content

Commit 5c084dd

Browse files
committed
Update on "Grab Current Tracing Fake Mode in a couple spots"
Fix for #99286. There were a couple locations we were instantiating new fake modes instead of grabbing the correct one from the current tracing context/inputs. cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx peterbell10 desertfire [ghstack-poisoned]
1 parent 88d4e72 commit 5c084dd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/_inductor/compile_fx.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,7 @@ def compile_fx_inner(
187187

188188
shape_env = _shape_env_from_inputs(example_inputs)
189189

190-
fake_mode = detect_fake_mode(example_inputs, ignore_tracing_context=True)
190+
fake_mode = detect_fake_mode(example_inputs)
191191
if not fake_mode:
192192
fake_mode = torch._subclasses.FakeTensorMode(allow_non_fake_inputs=True)
193193
FakeTensorProp(gm, mode=fake_mode).propagate(*example_inputs)

0 commit comments

Comments
 (0)