diff --git a/test/torch_npu_schema.json b/test/torch_npu_schema.json
index 44b695d355e0a0d5a8d9e0b8ae1d162ea5b78793..645df518f6b42dee5d48e738bc796a303025ea94 100644
--- a/test/torch_npu_schema.json
+++ b/test/torch_npu_schema.json
@@ -717,10 +717,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