from dataclasses import dataclass
from typing import List

from torch._export.serde.schema import Node


@dataclass
class ExternKernelNode:
    name: str
    node: Node


@dataclass
class ExternKernelNodes:
    nodes: List[ExternKernelNode]