[PIR] No.46 Migrate paddle.tile into pir (#57700)

zrr1999 · web-flow · commit 53e07afe2ada · 2023-09-26T18:56:53.000+08:00
diff --git a/paddle/fluid/pir/dialect/op_generator/vjp_interface_gen_op_list.py b/paddle/fluid/pir/dialect/op_generator/vjp_interface_gen_op_list.py
@@ -67,6 +67,7 @@
     'unsqueeze',
     'tril',
     'triu',
+    'tile',
 ]
 vjp_interface_implementation_gen_op_list = [
     'where',
@@ -112,4 +113,5 @@
     'unsqueeze',
     'tril',
     'triu',
+    'tile',
 ]
diff --git a/paddle/fluid/primitive/codegen/gen.py b/paddle/fluid/primitive/codegen/gen.py
@@ -41,6 +41,7 @@
     'where_grad',
     'tril_grad',
     'triu_grad',
+    'tile_grad',
     'tanh_grad',
     'mean_grad',
     'add_grad',
@@ -160,6 +161,7 @@
     'where_grad',
     'tril_grad',
     'triu_grad',
+    'tile_grad',
     'add_n',
     'mean',
     'sum',
diff --git a/python/paddle/tensor/manipulation.py b/python/paddle/tensor/manipulation.py
@@ -3168,7 +3168,7 @@ def tile(x, repeat_times, name=None):
             # Tensor(shape=[1, 6], dtype=int32, place=Place(gpu:0), stop_gradient=True,
             #        [[1, 2, 3, 1, 2, 3]])
     """
-    if in_dynamic_mode():
+    if in_dynamic_or_pir_mode():
         if isinstance(repeat_times, core.eager.Tensor):
             assert (
                 repeat_times.ndim == 1
diff --git a/test/legacy_test/test_tile_op.py b/test/legacy_test/test_tile_op.py
@@ -47,10 +47,10 @@ def init_data(self):
         self.repeat_times = [2]
 
     def test_check_output(self):
-        self.check_output(check_cinn=self.check_cinn)
+        self.check_output(check_cinn=self.check_cinn, check_new_ir=True)
 
     def test_check_grad(self):
-        self.check_grad(['X'], 'Out', check_prim=True)
+        self.check_grad(['X'], 'Out', check_prim=True, check_new_ir=True)
 
 
 class TestTileOpRank_ZeroDim1(TestTileOpRank1):
@@ -165,7 +165,7 @@ def init_data(self):
         self.infer_repeat_times = [-1]
 
     def test_check_output(self):
-        self.check_output()
+        self.check_output(check_new_ir=True)
 
     def test_check_grad(self):
         self.check_grad(['X'], 'Out')
@@ -206,7 +206,7 @@ def init_data(self):
         self.repeat_times = [2]
 
     def test_check_output(self):
-        self.check_output()
+        self.check_output(check_new_ir=True)
 
     def test_check_grad(self):
         self.check_grad(['X'], 'Out')
@@ -235,7 +235,7 @@ def if_enable_cinn(self):
         self.check_cinn = True
 
     def test_check_output(self):
-        self.check_output(check_cinn=self.check_cinn)
+        self.check_output(check_cinn=self.check_cinn, check_new_ir=True)
 
 
 class TestTileFP16OP(OpTest):
@@ -262,10 +262,10 @@ def init_data(self):
         self.repeat_times = [2, 1, 4]
 
     def test_check_output(self):
-        self.check_output(check_cinn=self.check_cinn)
+        self.check_output(check_cinn=self.check_cinn, check_new_ir=True)
 
     def test_check_grad(self):
-        self.check_grad(['X'], 'Out', check_prim=True)
+        self.check_grad(['X'], 'Out', check_prim=True, check_new_ir=True)
 
 
 @unittest.skipIf(
@@ -293,7 +293,9 @@ def if_enable_cinn(self):
 
     def test_check_output(self):
         place = core.CUDAPlace(0)
-        self.check_output_with_place(place, check_cinn=self.check_cinn)
+        self.check_output_with_place(
+            place, check_cinn=self.check_cinn, check_new_ir=True
+        )
 
     def init_data(self):
         self.dtype = np.uint16
@@ -302,7 +304,9 @@ def init_data(self):
 
     def test_check_grad(self):
         place = core.CUDAPlace(0)
-        self.check_grad_with_place(place, ['X'], 'Out', check_prim=True)
+        self.check_grad_with_place(
+            place, ['X'], 'Out', check_prim=True, check_new_ir=True
+        )
 
 
 # Situation 5: input x is Bool
@@ -320,7 +324,7 @@ def if_enable_cinn(self):
         self.check_cinn = True
 
     def test_check_output(self):
-        self.check_output(check_cinn=self.check_cinn)
+        self.check_output(check_cinn=self.check_cinn, check_new_ir=True)
 
 
 # Situation 56: input x is Integer
@@ -340,7 +344,7 @@ def if_enable_cinn(self):
         self.check_cinn = True
 
     def test_check_output(self):
-        self.check_output(check_cinn=self.check_cinn)
+        self.check_output(check_cinn=self.check_cinn, check_new_ir=True)
 
 
 class TestTileError(unittest.TestCase):

Original file line number	Diff line number	Diff line change
`@@ -67,6 +67,7 @@`
`67`	`67`	`'unsqueeze',`
`68`	`68`	`'tril',`
`69`	`69`	`'triu',`
	`70`	`+ 'tile',`
`70`	`71`	`]`
`71`	`72`	`vjp_interface_implementation_gen_op_list = [`
`72`	`73`	`'where',`
`@@ -112,4 +113,5 @@`
`112`	`113`	`'unsqueeze',`
`113`	`114`	`'tril',`
`114`	`115`	`'triu',`
	`116`	`+ 'tile',`
`115`	`117`	`]`