@@ -193,8 +193,8 @@ def _transform(self, *passes: PassType) -> "ExportedProgram":
193193 range_constraints = _get_updated_range_constraints (transformed_gm ),
194194 module_call_graph = copy .deepcopy (self ._module_call_graph ),
195195 example_inputs = self .example_inputs ,
196- verifier = self .verifier ,
197196 constants = self .constants ,
197+ verifiers = [self .verifier ],
198198 )
199199 transformed_ep .graph_module .meta .update (self .graph_module .meta )
200200 transformed_ep .graph_module .meta .update (res .graph_module .meta )
@@ -590,8 +590,8 @@ def _to_edge(ep, config: EdgeCompileConfig) -> "ExirExportedProgram":
590590 range_constraints = ep .exported_program .range_constraints ,
591591 module_call_graph = ep .exported_program .module_call_graph ,
592592 example_inputs = ep .exported_program .example_inputs ,
593- verifier = get_aten_verifier (enable = config ._check_ir_validity ),
594593 constants = ep .exported_program .constants ,
594+ verifiers = [get_aten_verifier (enable = config ._check_ir_validity )],
595595 ),
596596 False ,
597597 )
@@ -626,11 +626,13 @@ def _to_edge(ep, config: EdgeCompileConfig) -> "ExirExportedProgram":
626626 range_constraints = new_ep .exported_program .range_constraints ,
627627 module_call_graph = new_ep .exported_program .module_call_graph ,
628628 example_inputs = new_ep .exported_program .example_inputs ,
629- verifier = EXIREdgeDialectVerifier (
630- edge_compile_config = config ,
631- class_only = True ,
632- ),
633629 constants = new_ep .exported_program .constants ,
630+ verifiers = [
631+ EXIREdgeDialectVerifier (
632+ edge_compile_config = config ,
633+ class_only = True ,
634+ )
635+ ],
634636 )
635637 new_ep .after_to_edge_passes = True
636638 return new_ep
@@ -708,12 +710,14 @@ def _generate_edge_program(
708710 range_constraints = program .range_constraints ,
709711 module_call_graph = program .module_call_graph ,
710712 example_inputs = program .example_inputs ,
711- verifier = EXIREdgeDialectVerifier (
712- edge_compile_config = config ,
713- class_only = True ,
714- exception_list = ops_set_to_not_decompose ,
715- ),
716713 constants = program .constants ,
714+ verifiers = [
715+ EXIREdgeDialectVerifier (
716+ edge_compile_config = config ,
717+ class_only = True ,
718+ exception_list = ops_set_to_not_decompose ,
719+ )
720+ ],
717721 )
718722 # Lift the tensor constants created in ScalarToTensorPass
719723 edge_program = lift_constant_tensor_pass (edge_program )
0 commit comments