diff --git a/.github/labeler.yml b/.github/labeler.yml index bbdf14bc291..8b1acc77c26 100644 --- a/.github/labeler.yml +++ b/.github/labeler.yml @@ -116,7 +116,6 @@ "release notes: inductor (aoti)": - torch/_C/_aoti.pyi - torch/_dynamo/repro/aoti.py -- torch/_export/serde/aoti_schema.py - torch/_higher_order_ops/aoti_call_delegate.py - torch/_inductor/codegen/aoti_runtime/** - torch/_inductor/codegen/aoti_hipify_utils.py diff --git a/torch/_export/serde/aoti_schema.py b/torch/_export/serde/aoti_schema.py deleted file mode 100644 index d19add43705..00000000000 --- a/torch/_export/serde/aoti_schema.py +++ /dev/null @@ -1,14 +0,0 @@ -from dataclasses import dataclass - -from torch._export.serde.schema import Node - - -@dataclass -class ExternKernelNode: - name: str - node: Node - - -@dataclass -class ExternKernelNodes: - nodes: list[ExternKernelNode] diff --git a/torch/_inductor/extern_node_serializer.py b/torch/_inductor/extern_node_serializer.py index ffd39015203..0e5f42e7309 100644 --- a/torch/_inductor/extern_node_serializer.py +++ b/torch/_inductor/extern_node_serializer.py @@ -1,6 +1,6 @@ import json -from torch._export.serde.aoti_schema import ExternKernelNode, ExternKernelNodes, Node +from torch._export.serde.schema import ExternKernelNode, ExternKernelNodes, Node from torch._export.serde.serialize import _dataclass_to_dict, EnumEncoder from torch._inductor.ir import ExternKernelNode as inductor_ExternKernelNode