Update serialization and deserialization

akhilg-nv · akhilg-nv · commit 923cec59d0f8 · 2025-10-30T17:22:39.000-07:00
diff --git a/tripy/nvtripy/backend/api/executable.py b/tripy/nvtripy/backend/api/executable.py
@@ -396,6 +396,7 @@ def encode_executable(executable):
         "executable": base64.b64encode(executable._executable.serialize()).decode(),
         "_return_single_tensor_as_sequence": executable._return_single_tensor_as_sequence,
         "input_infos": executable.input_infos,
+        "leaf_names_by_arg": executable._leaf_names_by_arg,
     }
 
 
@@ -407,4 +408,5 @@ def decode_executable(executable_dict):
         executable_dict["arg_names"],
         return_single_tensor_as_sequence=executable_dict["_return_single_tensor_as_sequence"],
         input_infos=executable_dict["input_infos"],
+        leaf_names_by_arg=executable_dict.get("leaf_names_by_arg"),
     )

Original file line number	Diff line number	Diff line change
`@@ -396,6 +396,7 @@ def encode_executable(executable):`
`396`	`396`	`"executable": base64.b64encode(executable._executable.serialize()).decode(),`
`397`	`397`	`"_return_single_tensor_as_sequence": executable._return_single_tensor_as_sequence,`
`398`	`398`	`"input_infos": executable.input_infos,`
	`399`	`+ "leaf_names_by_arg": executable._leaf_names_by_arg,`
`399`	`400`	`}`
`400`	`401`
`401`	`402`
`@@ -407,4 +408,5 @@ def decode_executable(executable_dict):`
`407`	`408`	`executable_dict["arg_names"],`
`408`	`409`	`return_single_tensor_as_sequence=executable_dict["_return_single_tensor_as_sequence"],`
`409`	`410`	`input_infos=executable_dict["input_infos"],`
	`411`	`+ leaf_names_by_arg=executable_dict.get("leaf_names_by_arg"),`
`410`	`412`	`)`