diff --git a/paddle/fluid/framework/paddle2cinn/cinn_zero_tensor_trick_pass.cc b/paddle/fluid/framework/paddle2cinn/cinn_zero_tensor_trick_pass.cc index 3ff18200a0d92d426dd5466ee67e88b20d856a93..a7c06af47244eb5922bd4955214ce0f4ff1dd3d6 100644 --- a/paddle/fluid/framework/paddle2cinn/cinn_zero_tensor_trick_pass.cc +++ b/paddle/fluid/framework/paddle2cinn/cinn_zero_tensor_trick_pass.cc @@ -58,11 +58,26 @@ void CinnZeroTensorTrickPass::ApplyImpl(ir::Graph* graph) const { } } - // CINN ops in this white list support 0D-Tensor + // CINN ops in this white list support 0D-Tensor, wait-list = {"remainder"} const std::unordered_set white_op_list{"elementwise_add", "elementwise_sub", "elementwise_mul", - "elementwise_div"}; + "elementwise_div", + "atan2", + "floor_divide", + "mod", + "maximum", + "minimum", + "pow", + "logical_and", + "logical_or", + "logical_xor", + "equal", + "not_equal", + "greater_than", + "greater_equal", + "less_than", + "less_equal"}; std::unordered_set white_tensor_name; // enable white_op_list only when graph_node_size = 1, which means single op // test