[CINN]Fix topk indices no grad problem (#2935)

PaddlePaddle · Sep 5, 2024 · dfa3642 · dfa3642
1 parent 71828c0
commit dfa3642
Show file tree

Hide file tree

Showing 16 changed files with 48 additions and 0 deletions.
diff --git a/...e/sublayer1000/Det_cases/cascade_rcnn_cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco/SIR_85.py b/...e/sublayer1000/Det_cases/cascade_rcnn_cascade_mask_rcnn_r50_vd_fpn_ssld_1x_coco/SIR_85.py
@@ -25,6 +25,9 @@ def forward(
         var_11 = paddle.tensor.search.where(var_9, var_10, var_8)
         var_12 = var_2.flatten()
         var_13 = var_11.flatten()
+        # bool/int tensors has no grad
+        var_12.stop_gradient = True
+        var_13.stop_gradient = True
         return var_12, var_13
 
 

diff --git a/...LT_new/layercase/sublayer1000/Det_cases/dcn_faster_rcnn_dcn_r101_vd_fpn_1x_coco/SIR_49.py b/...LT_new/layercase/sublayer1000/Det_cases/dcn_faster_rcnn_dcn_r101_vd_fpn_1x_coco/SIR_49.py
@@ -34,6 +34,9 @@ def forward(
         var_20 = paddle.tensor.search.where(var_18, var_19, var_11)
         var_21 = var_2.flatten()
         var_22 = var_20.flatten()
+        # bool/int tensors has no grad
+        var_21.stop_gradient = True
+        var_22.stop_gradient = True
         return var_21, var_22
 
 

diff --git a/...T_new/layercase/sublayer1000/Det_cases/faster_rcnn_faster_rcnn_r101_fpn_2x_coco/SIR_47.py b/...T_new/layercase/sublayer1000/Det_cases/faster_rcnn_faster_rcnn_r101_fpn_2x_coco/SIR_47.py
@@ -34,6 +34,9 @@ def forward(
         var_20 = paddle.tensor.search.where(var_18, var_19, var_11)
         var_21 = var_2.flatten()
         var_22 = var_20.flatten()
+        # bool/int tensors has no grad
+        var_21.stop_gradient = True
+        var_22.stop_gradient = True
         return var_21, var_22
 
 

diff --git a/...ew/layercase/sublayer1000/Det_cases/faster_rcnn_faster_rcnn_r101_vd_fpn_1x_coco/SIR_49.py b/...ew/layercase/sublayer1000/Det_cases/faster_rcnn_faster_rcnn_r101_vd_fpn_1x_coco/SIR_49.py
@@ -34,6 +34,9 @@ def forward(
         var_20 = paddle.tensor.search.where(var_18, var_19, var_11)
         var_21 = var_2.flatten()
         var_22 = var_20.flatten()
+        # bool/int tensors has no grad
+        var_21.stop_gradient = True
+        var_22.stop_gradient = True
         return var_21, var_22
 
 

diff --git a/...new/layercase/sublayer1000/Det_cases/faster_rcnn_faster_rcnn_r50_vd_fpn_2x_coco/SIR_49.py b/...new/layercase/sublayer1000/Det_cases/faster_rcnn_faster_rcnn_r50_vd_fpn_2x_coco/SIR_49.py
@@ -34,6 +34,9 @@ def forward(
         var_20 = paddle.tensor.search.where(var_18, var_19, var_11)
         var_21 = var_2.flatten()
         var_22 = var_20.flatten()
+        # bool/int tensors has no grad
+        var_21.stop_gradient = True
+        var_22.stop_gradient = True
         return var_21, var_22
 
 

diff --git a/...ayercase/sublayer1000/Det_cases/faster_rcnn_faster_rcnn_r50_vd_fpn_ssld_1x_coco/SIR_49.py b/...ayercase/sublayer1000/Det_cases/faster_rcnn_faster_rcnn_r50_vd_fpn_ssld_1x_coco/SIR_49.py
@@ -34,6 +34,9 @@ def forward(
         var_20 = paddle.tensor.search.where(var_18, var_19, var_11)
         var_21 = var_2.flatten()
         var_22 = var_20.flatten()
+        # bool/int tensors has no grad
+        var_21.stop_gradient = True
+        var_22.stop_gradient = True
         return var_21, var_22
 
 

diff --git a/...layercase/sublayer1000/Det_cases/faster_rcnn_faster_rcnn_swin_tiny_fpn_2x_coco/SIR_114.py b/...layercase/sublayer1000/Det_cases/faster_rcnn_faster_rcnn_swin_tiny_fpn_2x_coco/SIR_114.py
@@ -25,6 +25,9 @@ def forward(
         var_11 = paddle.tensor.search.where(var_9, var_10, var_8)
         var_12 = var_2.flatten()
         var_13 = var_11.flatten()
+        # bool/int tensors has no grad
+        var_12.stop_gradient = True
+        var_13.stop_gradient = True
         return var_12, var_13
 
 

diff --git a/.../layercase/sublayer1000/Det_cases/faster_rcnn_faster_rcnn_swin_tiny_fpn_2x_coco/SIR_99.py b/.../layercase/sublayer1000/Det_cases/faster_rcnn_faster_rcnn_swin_tiny_fpn_2x_coco/SIR_99.py
@@ -34,6 +34,9 @@ def forward(
         var_20 = paddle.tensor.search.where(var_18, var_19, var_11)
         var_21 = var_2.flatten()
         var_22 = var_20.flatten()
+        # bool/int tensors has no grad
+        var_21.stop_gradient = True
+        var_22.stop_gradient = True
         return var_21, var_22
 
 

diff --git a/...eLT_new/layercase/sublayer1000/Det_cases/hrnet_faster_rcnn_hrnetv2p_w18_1x_coco/SIR_43.py b/...eLT_new/layercase/sublayer1000/Det_cases/hrnet_faster_rcnn_hrnetv2p_w18_1x_coco/SIR_43.py
@@ -34,6 +34,9 @@ def forward(
         var_20 = paddle.tensor.search.where(var_18, var_19, var_11)
         var_21 = var_2.flatten()
         var_22 = var_20.flatten()
+        # bool/int tensors has no grad
+        var_21.stop_gradient = True
+        var_22.stop_gradient = True
         return var_21, var_22
 
 

diff --git a/...eLT_new/layercase/sublayer1000/Det_cases/hrnet_faster_rcnn_hrnetv2p_w18_2x_coco/SIR_58.py b/...eLT_new/layercase/sublayer1000/Det_cases/hrnet_faster_rcnn_hrnetv2p_w18_2x_coco/SIR_58.py
@@ -25,6 +25,9 @@ def forward(
         var_11 = paddle.tensor.search.where(var_9, var_10, var_8)
         var_12 = var_2.flatten()
         var_13 = var_11.flatten()
+        # bool/int tensors has no grad
+        var_12.stop_gradient = True
+        var_13.stop_gradient = True
         return var_12, var_13
 
 

diff --git a/.../layercase/sublayer1000/Det_cases/mask_rcnn_mask_rcnn_x101_vd_64x4d_fpn_2x_coco/SIR_49.py b/.../layercase/sublayer1000/Det_cases/mask_rcnn_mask_rcnn_x101_vd_64x4d_fpn_2x_coco/SIR_49.py
@@ -34,6 +34,9 @@ def forward(
         var_20 = paddle.tensor.search.where(var_18, var_19, var_11)
         var_21 = var_2.flatten()
         var_22 = var_20.flatten()
+        # bool/int tensors has no grad
+        var_21.stop_gradient = True
+        var_22.stop_gradient = True
         return var_21, var_22
 
 

diff --git a/...T_new/layercase/sublayer1000/Det_cases/rcnn_enhance_faster_rcnn_enhance_3x_coco/SIR_49.py b/...T_new/layercase/sublayer1000/Det_cases/rcnn_enhance_faster_rcnn_enhance_3x_coco/SIR_49.py
@@ -34,6 +34,9 @@ def forward(
         var_20 = paddle.tensor.search.where(var_18, var_19, var_11)
         var_21 = var_2.flatten()
         var_22 = var_20.flatten()
+        # bool/int tensors has no grad
+        var_21.stop_gradient = True
+        var_22.stop_gradient = True
         return var_21, var_22
 
 

diff --git a/.../layercase/sublayer1000/Det_cases/sniper_faster_rcnn_r50_fpn_1x_sniper_visdrone/SIR_47.py b/.../layercase/sublayer1000/Det_cases/sniper_faster_rcnn_r50_fpn_1x_sniper_visdrone/SIR_47.py
@@ -34,6 +34,9 @@ def forward(
         var_20 = paddle.tensor.search.where(var_18, var_19, var_11)
         var_21 = var_2.flatten()
         var_22 = var_20.flatten()
+        # bool/int tensors has no grad
+        var_21.stop_gradient = True
+        var_22.stop_gradient = True
         return var_21, var_22
 
 

diff --git a/.../layercase/sublayer1000/Det_cases/sniper_faster_rcnn_r50_fpn_1x_sniper_visdrone/SIR_62.py b/.../layercase/sublayer1000/Det_cases/sniper_faster_rcnn_r50_fpn_1x_sniper_visdrone/SIR_62.py
@@ -25,6 +25,9 @@ def forward(
         var_11 = paddle.tensor.search.where(var_9, var_10, var_8)
         var_12 = var_2.flatten()
         var_13 = var_11.flatten()
+        # bool/int tensors has no grad
+        var_12.stop_gradient = True
+        var_13.stop_gradient = True
         return var_12, var_13
 
 

diff --git a/...eLT_new/layercase/sublayer1000/Det_cases/sniper_faster_rcnn_r50_fpn_1x_visdrone/SIR_47.py b/...eLT_new/layercase/sublayer1000/Det_cases/sniper_faster_rcnn_r50_fpn_1x_visdrone/SIR_47.py
@@ -34,6 +34,9 @@ def forward(
         var_20 = paddle.tensor.search.where(var_18, var_19, var_11)
         var_21 = var_2.flatten()
         var_22 = var_20.flatten()
+        # bool/int tensors has no grad
+        var_21.stop_gradient = True
+        var_22.stop_gradient = True
         return var_21, var_22
 
 

diff --git a/...eLT_new/layercase/sublayer1000/Det_cases/sniper_faster_rcnn_r50_fpn_1x_visdrone/SIR_62.py b/...eLT_new/layercase/sublayer1000/Det_cases/sniper_faster_rcnn_r50_fpn_1x_visdrone/SIR_62.py
@@ -25,6 +25,9 @@ def forward(
         var_11 = paddle.tensor.search.where(var_9, var_10, var_8)
         var_12 = var_2.flatten()
         var_13 = var_11.flatten()
+        # bool/int tensors has no grad
+        var_12.stop_gradient = True
+        var_13.stop_gradient = True
         return var_12, var_13