process makefile

python/paddle/decomposition/rules.py

@@ -20,8 +20,7 @@
 def mean(x, axis, keepdim):
     """define composite rule of op mean"""
     x_shape = x.shape
-    axes = axis or tuple(range(0, len(x_shape)))
-    axes = (axes,) if isinstance(axes, int) else axes
+    axes = (axis,) if isinstance(axis, int) else tuple(range(0, len(x_shape)))
     sum_x = sum(x, axis=axes, keepdim=keepdim)
     value_to_fill = 1
     for axis in axes:

python/paddle/incubate/autograd/composite_rules.py

@@ -171,11 +171,11 @@ def layernorm_composite(x, scale, bias, epsilon, begin_norm_axis):
     out = difference * rsqrt_var
 
     if scale is not None:
-        if x.shape[begin_norm_axis:] is not scale.shape:
+        if x.shape[begin_norm_axis:] != scale.shape:
             scale = reshape(scale, x.shape[begin_norm_axis:])
         out = out * scale
     if bias is not None:
-        if x.shape[begin_norm_axis:] is not bias.shape:
+        if x.shape[begin_norm_axis:] != bias.shape:
             bias = reshape(bias, x.shape[begin_norm_axis:])
         out = out + bias
 
@@ -266,8 +266,7 @@ def mean_composite(x, axis, keepdim):
         is_amp = True
         x = cast(x, "float32")
 
-    axes = axis or list(range(0, len(x.shape)))
-    axes = [axes] if isinstance(axes, int) else axes
+    axes = [axis] if isinstance(axis, int) else list(range(0, len(x.shape)))
     sum_x = sum(x, axis=axes, keepdim=keepdim)
     ele_nums_list = [x.shape[axis] for axis in axes]
     if ele_nums_list == []:

test/prim/new_ir_prim/CMakeLists.txt

@@ -1,10 +1,20 @@
+set(TEST_PRIM_PURE_NEW_IR_CASES test_prim_program)
+
+foreach(target ${TEST_PRIM_PURE_NEW_IR_CASES})
+  py_test_modules(${target} MODULES ${target} ENVS GLOG_v=1
+                  FLAGS_enable_new_ir_api=true)
+endforeach()
+
 file(
-  GLOB TEST_INTERP_CASES
+  GLOB TEST_PRIM_TRANS_NEW_IR_CASES
   RELATIVE "${CMAKE_CURRENT_SOURCE_DIR}"
   "test_*.py")
-string(REPLACE ".py" "" TEST_INTERP_CASES "${TEST_INTERP_CASES}")
+string(REPLACE ".py" "" TEST_PRIM_TRANS_NEW_IR_CASES
+               "${TEST_PRIM_TRANS_NEW_IR_CASES}")
+
+list(REMOVE_ITEM TEST_PRIM_TRANS_NEW_IR_CASES ${TEST_PRIM_PURE_NEW_IR_CASES})
 
-foreach(target ${TEST_INTERP_CASES})
+foreach(target ${TEST_PRIM_TRANS_NEW_IR_CASES})
   py_test_modules(${target} MODULES ${target} ENVS GLOG_v=1
                   FLAGS_enable_new_ir_in_executor=true)
 endforeach()