diff --git a/test/torch_npu_schema.json b/test/torch_npu_schema.json
index 59b67759ddf15b23a732eab3f4eaad48dcebd70a..af08554dc68338f3b7463ed1d78021b6e6f2e4d6 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 6e3ef8386ea9ef68573395bd2f4e9ed01cc9a388..37fd0548aebd8198c745a2a6a748eaee20c0f878 160000
--- a/third_party/torchair/torchair
+++ b/third_party/torchair/torchair
@@ -1 +1 @@
-Subproject commit 6e3ef8386ea9ef68573395bd2f4e9ed01cc9a388
+Subproject commit 37fd0548aebd8198c745a2a6a748eaee20c0f878