[CINN] Add PRIM_FORWARD_BLOCKLIST in sub_graph test (#69398)

* add prim_blacklist * delete note
PaddlePaddle · Nov 15, 2024 · 4043b8c · 4043b8c
1 parent 52f55e1
commit 4043b8c
Show file tree

Hide file tree

Showing 5 changed files with 2 additions and 12 deletions.
diff --git a/test/ir/pir/cinn/sub_graphs/CMakeLists.txt b/test/ir/pir/cinn/sub_graphs/CMakeLists.txt
@@ -1,4 +1,5 @@
 if(WITH_GPU)
+  set(PRIM_FORWARD_BLOCKLIST "pd_op.dropout")
 
   file(
     GLOB DYNAMIC_BUILD_TESTS
@@ -13,6 +14,7 @@ if(WITH_GPU)
         ${CMAKE_COMMAND} -E env
         PYTHONPATH=${CMAKE_BINARY_DIR}:${CMAKE_BINARY_DIR}/python/:$ENV{PYTHONPATH}
         FLAGS_prim_enable_dynamic=1 FLAGS_check_infer_symbolic=1
+        FLAGS_prim_forward_blacklist=${PRIM_FORWARD_BLOCKLIST}
         FLAGS_cudnn_deterministic=true ${PYTHON_EXECUTABLE}
         ${CMAKE_CURRENT_SOURCE_DIR}/${cinn_sub_graph_test_name}.py
       WORKING_DIRECTORY ${CMAKE_BINARY_DIR})

diff --git a/test/ir/pir/cinn/sub_graphs/test_sub_graph_18.py b/test/ir/pir/cinn/sub_graphs/test_sub_graph_18.py
@@ -68,10 +68,6 @@ def init(self):
             paddle.rand(shape=[22, 1536, 8, 8], dtype=paddle.float32),
         )
         self.net = LayerCase
-        self.with_precision_compare = False
-        self.with_train = False
-
-    # NOTE output mismatch with prim
 
 
 if __name__ == '__main__':

diff --git a/test/ir/pir/cinn/sub_graphs/test_sub_graph_25.py b/test/ir/pir/cinn/sub_graphs/test_sub_graph_25.py
@@ -68,8 +68,6 @@ def init(self):
             paddle.rand(shape=[11, 1280, 7, 7], dtype=paddle.float32),
         )
         self.net = LayerCase
-        self.with_precision_compare = False
-        self.with_train = False
 
 
 if __name__ == '__main__':

diff --git a/test/ir/pir/cinn/sub_graphs/test_sub_graph_28.py b/test/ir/pir/cinn/sub_graphs/test_sub_graph_28.py
@@ -68,10 +68,6 @@ def init(self):
             paddle.rand(shape=[10, 320, 8, 8], dtype=paddle.float32),
         )
         self.net = LayerCase
-        self.with_precision_compare = False
-        self.with_train = False
-
-    # NOTE prim + cinn lead to error
 
 
 if __name__ == '__main__':

diff --git a/test/ir/pir/cinn/sub_graphs/test_sub_graph_29.py b/test/ir/pir/cinn/sub_graphs/test_sub_graph_29.py
@@ -68,8 +68,6 @@ def init(self):
             paddle.rand(shape=[10, 2048, 10, 10], dtype=paddle.float32),
         )
         self.net = LayerCase
-        self.with_precision_compare = False
-        self.with_train = False
 
 
 if __name__ == '__main__':