Skip to content

Commit f07d2db

Browse files
committed
add support for fx.Proxy
1 parent 4b8160a commit f07d2db

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

onnx_diagnostic/helpers/helper.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -801,6 +801,9 @@ def _torch_sym_int_to_str(value: "torch.SymInt") -> Union[int, str]: # noqa: F
801801
print(f"[string_type] TT8:{type(obj)}")
802802
return repr(obj).replace(" ", "").replace("\n", " ")
803803

804+
if isinstance(obj, torch.fx.proxy.Proxy):
805+
return repr(obj)
806+
804807
if ignore:
805808
if verbose:
806809
print(f"[string_type] CACHE4:{type(obj)}")

0 commit comments

Comments
 (0)