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):