From d3c50c1d045d2598e925a3a3b9859979f01f6234 Mon Sep 17 00:00:00 2001 From: Zhengxu Chen Date: Fri, 10 May 2024 11:08:56 -0700 Subject: [PATCH] explicitly import _export.exported_program. (#3574) Summary: We will still merge two serialization, but for now just fix the missing imports. Reviewed By: tugsbayasgalan Differential Revision: D57219425 --- exir/serde/export_serialize.py | 1 + 1 file changed, 1 insertion(+) diff --git a/exir/serde/export_serialize.py b/exir/serde/export_serialize.py index 87691dfbee2..13590b78dde 100644 --- a/exir/serde/export_serialize.py +++ b/exir/serde/export_serialize.py @@ -22,6 +22,7 @@ import sympy import torch +import torch._export.exported_program import torch.export.exported_program as ep from torch._export.serde.schema import (