From 56a66b93e381bf5a561710bba55aa6d9cba8ac78 Mon Sep 17 00:00:00 2001 From: l00800044 Date: Thu, 11 Sep 2025 11:40:07 +0800 Subject: [PATCH] update torchair custom_op and commitid master --- test/torch_npu_schema.json | 4 ++-- third_party/torchair/torchair | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/test/torch_npu_schema.json b/test/torch_npu_schema.json index 6b2fc42f85..cbf51e4eb6 100644 --- a/test/torch_npu_schema.json +++ b/test/torch_npu_schema.json @@ -675,10 +675,10 @@ "signature": "(self)" }, "torch_npu.dynamo.torchair.ge.custom_op": { - "signature": "(op_type: str, *, inputs: Optional[Dict[str, Union[ForwardRef('Tensor'), List[ForwardRef('Tensor')], NoneType]]], outputs: Optional[List[Union[str, Tuple[str, int]]]], attrs: Optional[Dict[str, ForwardRef('_Attr')]] = None, node_name: Optional[str] = None)" + "signature": "(op_type: str, *args, inputs: Optional[Dict[str, Union[ForwardRef('Tensor'), List[ForwardRef('Tensor')], NoneType]]] = None, outputs: Optional[List[Union[str, Tuple[str, int]]]] = None, attrs: Optional[Dict[str, ForwardRef('_Attr')]] = None, node_name: Optional[str] = None)" }, "torch_npu.dynamo.torchair.ge.ge_custom.custom_op": { - "signature": "(op_type: str, *, inputs: Optional[Dict[str, Union[ForwardRef('Tensor'), List[ForwardRef('Tensor')], NoneType]]], outputs: Optional[List[Union[str, Tuple[str, int]]]], attrs: Optional[Dict[str, ForwardRef('_Attr')]] = None, node_name: Optional[str] = None)" + "signature": "(op_type: str, *args, inputs: Optional[Dict[str, Union[ForwardRef('Tensor'), List[ForwardRef('Tensor')], NoneType]]] = None, outputs: Optional[List[Union[str, Tuple[str, int]]]] = None, attrs: Optional[Dict[str, ForwardRef('_Attr')]] = None, node_name: Optional[str] = None)" }, "torch_npu.dynamo.torchair.inference.cache_compile": { "signature": "(func, *, config: Optional[torchair.configs.compiler_config.CompilerConfig] = None, dynamic: bool = True, cache_dir: Optional[str] = None, global_rank: Optional[int] = None, tp_rank: Optional[int] = None, pp_rank: Optional[int] = None, **kwargs) -> Callable" diff --git a/third_party/torchair/torchair b/third_party/torchair/torchair index 6e3ef8386e..37fd0548ae 160000 --- a/third_party/torchair/torchair +++ b/third_party/torchair/torchair @@ -1 +1 @@ -Subproject commit 6e3ef8386ea9ef68573395bd2f4e9ed01cc9a388 +Subproject commit 37fd0548aebd8198c745a2a6a748eaee20c0f878 -- Gitee