reorganize the prim unittests struct (#49679)

python/paddle/fluid/tests/unittests/prim/CMakeLists.txt

@@ -3,10 +3,10 @@ file(
   RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}"
   "test_*.py")
 string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")
-set(GC_ENVS FLAGS_eager_delete_tensor_gb=0.0)
 
 foreach(TEST_OP ${TEST_OPS})
   py_test_modules(${TEST_OP} MODULES ${TEST_OP} ENVS ${GC_ENVS})
 endforeach()
 
-add_subdirectory(api)
+add_subdirectory(comp)
+add_subdirectory(prim)

python/paddle/fluid/tests/unittests/prim/README.txt

+# Prim
+
+Test composite and primitive API and related autodiff rules.
+
+
+## Package Structure
+
+- comp(abbr for composite):
+
+  Test composite API which is composed of primitive api, but not include the autodiff rules for primitive api.
+
+- prim(abbr for primitive):
+
+  Test primitive api and autodiff rules.
+
+    - vjp: Test vjp rules.
+
+        - eager: Test vjp rules in eager mode.
+        - static: Test vjp rules in static mode.
+
+    - jvp(TODO): Test jvp rules.
+
+
+## How to
+
+- Forward API and First-rder Autodiff
+
+  Compare numerical value with raw phi operators using tools such as `np.testing.assert_allclose`
+
+
+- Higher-order Autodiff
+
+  Compare numerical value with the result computed by finite difference. Tool
+  used for computing Higher-order finite difference will be provided in the future.

python/paddle/fluid/tests/unittests/prim/api/prim/CMakeLists.txt → python/paddle/fluid/tests/unittests/prim/comp/CMakeLists.txt

@@ -3,7 +3,6 @@ file(
   RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}"
   "test_*.py")
 string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")
-set(GC_ENVS FLAGS_eager_delete_tensor_gb=0.0)
 
 foreach(TEST_OP ${TEST_OPS})
   py_test_modules(${TEST_OP} MODULES ${TEST_OP} ENVS ${GC_ENVS})

python/paddle/fluid/tests/unittests/prim/api/CMakeLists.txt → python/paddle/fluid/tests/unittests/prim/prim/CMakeLists.txt

@@ -3,11 +3,9 @@ file(
   RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}"
   "test_*.py")
 string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")
-set(GC_ENVS FLAGS_eager_delete_tensor_gb=0.0)
 
 foreach(TEST_OP ${TEST_OPS})
   py_test_modules(${TEST_OP} MODULES ${TEST_OP} ENVS ${GC_ENVS})
 endforeach()
 
-add_subdirectory(comp)
-add_subdirectory(prim)
+add_subdirectory(vjp)

python/paddle/fluid/tests/unittests/prim/api/comp/CMakeLists.txt → python/paddle/fluid/tests/unittests/prim/prim/vjp/CMakeLists.txt

@@ -3,11 +3,10 @@ file(
   RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}"
   "test_*.py")
 string(REPLACE ".py" "" TEST_OPS "${TEST_OPS}")
-set(GC_ENVS FLAGS_eager_delete_tensor_gb=0.0)
 
 foreach(TEST_OP ${TEST_OPS})
   py_test_modules(${TEST_OP} MODULES ${TEST_OP} ENVS ${GC_ENVS})
 endforeach()
 
-add_subdirectory(eager_test)
-add_subdirectory(static_test)
+add_subdirectory(eager)
+add_subdirectory(static)