From 26b85eadde28645c9b04b2d5a5b37f4d810b5100 Mon Sep 17 00:00:00 2001 From: Thiago Crepaldi Date: Sat, 24 Feb 2024 03:33:35 -0800 Subject: [PATCH] Ignore .numpy() under FakeTensorMode() (#120261) Summary: Fixes #120259 X-link: https://github.com/pytorch/pytorch/pull/120261 Approved by: https://github.com/jansel Reviewed By: jeanschmidt Differential Revision: D54123967 fbshipit-source-id: 1e41321385fc5f8924e26775ddf6a6292c2ad6c6 --- userbenchmark/dynamo/dynamobench/_dynamo/utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py index 933cabc5a2..7432861920 100644 --- a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py +++ b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py @@ -2013,6 +2013,8 @@ def nnmodule_has_hooks( def to_numpy_helper(value): """Convert tensor and tnp.ndarray to numpy.ndarray.""" + if is_fake(value): + return value if isinstance(value, tnp.ndarray): return to_numpy_helper(value.tensor) elif isinstance(value, torch.Tensor):