Removed deprecated dialect field from EP schema.

Summary: As noted in the serialization schema, we've deprecated "dialect" field from EP and we're trying to remove it in this diff.

Differential Revision: D59696447

Author: zhxchen17

exir/serde/export_serialize.py

@@ -1063,12 +1063,13 @@ def serialize(self, exported_program: ep.ExportedProgram) -> SerializedArtifact:
             additional_kwargs["verifiers"] = [
                 v.dialect for v in exported_program.verifiers
             ]
+        elif hasattr(exported_program, "dialect"):
+            additional_kwargs["dialect"] = exported_program.dialect
         serialized_ep = ExportedProgram(
             graph_module=serialized_graph_module,
             opset_version=self.opset_version,
             range_constraints=serialized_range_constraints,
             schema_version=SchemaVersion(-1, -1),
-            dialect=exported_program.dialect,
             **additional_kwargs,
         )
 

exir/serde/serialize.py

@@ -349,13 +349,14 @@ def serialize(
             additional_kwargs["verifiers"] = [
                 v.dialect for v in exported_program.verifiers
             ]
+        elif hasattr(exported_program, "dialect"):
+            additional_kwargs["dialect"] = exported_program.dialect
         return export_serialize.SerializedArtifact(
             schema.ExportedProgram(
                 graph_module=serialized_graph_module,
                 opset_version=self.opset_version,
                 range_constraints=serialized_range_constraints,
                 schema_version=SchemaVersion(-1, -1),
-                dialect=exported_program.dialect,
                 **additional_kwargs,
             ),
             export_serialize.serialize_torch_artifact(exported_program.state_dict),
@@ -681,16 +682,22 @@ def deserialize(
 
         dummy_g = torch.fx.Graph()
         dummy_g.output(())
+        serialized_ep = serialized_artifact.exported_program
+        additional_kwargs = {}
+        if hasattr(serialized_ep, "verifiers"):
+            additional_kwargs["verifiers"] = [
+                load_verifier(v) for v in serialized_ep.verifiers  # pyre-ignore
+            ]
+        elif hasattr(serialized_ep, "dialect"):
+            additional_kwargs["verifier"] = load_verifier(serialized_ep.dialect)  # pyre-ignore
         exported_program = exir.ExportedProgram(
             root=state_dict,
             graph=dummy_g,
             graph_signature=ep.ExportGraphSignature(input_specs=[], output_specs=[]),
             state_dict=state_dict,  # TODO(T157676982)
             range_constraints=range_constraints,
             module_call_graph=module_call_graph,
-            verifier=load_verifier(
-                serialized_artifact.exported_program.dialect  # pyre-ignore
-            ),
+            **additional_kwargs,
         )
         exported_program.graph_module.graph = graph_module.graph
         exported_program._graph_signature = res.signature