fix

Author: HydrogenSulfate

python/paddle/base/dygraph/math_op_patch.py

@@ -296,6 +296,7 @@ def _new_full_(
         dtype: DTypeLike | None = None,
         device: PlaceLike | None = None,
         requires_grad: bool = False,
+        pin_memory: bool = False,
     ) -> Tensor:
         if dtype is None:
             dtype = var.dtype
@@ -308,6 +309,7 @@ def _new_full_(
             dtype=dtype,
             device=device,
             requires_grad=requires_grad,
+            pin_memory=pin_memory,
         )
 
     def _new_empty_(
@@ -339,6 +341,7 @@ def _new_ones_(
         dtype: DTypeLike | None = None,
         device: PlaceLike | None = None,
         requires_grad: bool = False,
+        pin_memory: bool = False,
     ) -> Tensor:
         if dtype is None:
             dtype = var.dtype
@@ -351,6 +354,7 @@ def _new_ones_(
             dtype,
             device=device,
             requires_grad=requires_grad,
+            pin_memory=pin_memory,
         )
 
     def _new_zeros_(
@@ -360,6 +364,7 @@ def _new_zeros_(
         dtype: DTypeLike | None = None,
         device: PlaceLike | None = None,
         requires_grad: bool = False,
+        pin_memory: bool = False,
     ) -> Tensor:
         if dtype is None:
             dtype = var.dtype
@@ -372,6 +377,7 @@ def _new_zeros_(
             dtype,
             device=device,
             requires_grad=requires_grad,
+            pin_memory=pin_memory,
         )
 
     @property

python/paddle/pir/math_op_patch.py

@@ -647,6 +647,7 @@ def _new_full_(
         dtype: DTypeLike | None = None,
         device: PlaceLike | None = None,
         requires_grad: bool = False,
+        pin_memory: bool = False,
     ):
         """
 
@@ -682,6 +683,7 @@ def _new_full_(
             dtype=dtype,
             device=device,
             requires_grad=requires_grad,
+            pin_memory=pin_memory,
         )
 
     def _new_empty_(
@@ -736,6 +738,7 @@ def _new_ones_(
         dtype: DTypeLike | None = None,
         device: PlaceLike | None = None,
         requires_grad: bool = False,
+        pin_memory: bool = False,
     ):
         """
 
@@ -771,6 +774,7 @@ def _new_ones_(
             dtype=dtype,
             device=device,
             requires_grad=requires_grad,
+            pin_memory=pin_memory,
         )
 
     def _new_zeros_(
@@ -780,6 +784,7 @@ def _new_zeros_(
         dtype: DTypeLike | None = None,
         device: PlaceLike | None = None,
         requires_grad: bool = False,
+        pin_memory: bool = False,
     ):
         """
 
@@ -815,6 +820,7 @@ def _new_zeros_(
             dtype=dtype,
             device=device,
             requires_grad=requires_grad,
+            pin_memory=pin_memory,
         )
 
     def _int_(self):