File tree Expand file tree Collapse file tree 2 files changed +11
-3
lines changed
backends/xnnpack/test/tester Expand file tree Collapse file tree 2 files changed +11
-3
lines changed Original file line number Diff line number Diff line change @@ -152,7 +152,7 @@ def run(
152
152
) -> None :
153
153
prepared = prepare_pt2e (artifact .exported_program .graph_module , self .quantizer )
154
154
converted = convert_pt2e (prepared )
155
- artifact .exported_program .graph_module = converted
155
+ artifact .exported_program ._graph_module = converted
156
156
self .converted_program = artifact
157
157
158
158
@property
Original file line number Diff line number Diff line change @@ -287,8 +287,16 @@ def to_backend(
287
287
tagged_graph_module , partitioner_instance , edge_program
288
288
)
289
289
290
- edge_program .graph_module = tagged_graph_module
291
- return edge_program
290
+ return ExportedProgram (
291
+ tagged_graph_module ,
292
+ tagged_graph_module .graph ,
293
+ copy .deepcopy (edge_program .graph_signature ),
294
+ copy .deepcopy (edge_program .call_spec ),
295
+ edge_program .state_dict ,
296
+ copy .deepcopy (edge_program .range_constraints ),
297
+ copy .deepcopy (edge_program .equality_constraints ),
298
+ copy .deepcopy (edge_program .module_call_graph ),
299
+ )
292
300
293
301
294
302
def to_backend_multiple (
You can’t perform that action at this time.
0 commit comments