Exception API

PADDLE_ENFORCE

使用方式：

 PADDLE_ENFORCE_{TYPE}(cond_a, // 条件A
                       cond_b, // 条件B, 根据TYPE可选
                       phi::errors::{ERR_TYPE}("{ERR_MSG}"));

根据TYPE的不同，分为：

Exception宏 判断条件 报错信息
PADDLE_ENFORCE_EQ cond_a == cond_b 触发ERR_TYPE异常和报ERR_MSG
PADDLE_ENFORCE_NE cond_a != cond_b 触发ERR_TYPE异常和报ERR_MSG
PADDLE_ENFORCE_GT cond_a > cond_b 触发ERR_TYPE异常和报ERR_MSG
PADDLE_ENFORCE_GE cond_a >= cond_b 触发ERR_TYPE异常和报ERR_MSG
PADDLE_ENFORCE_LT cond_a < cond_b 触发ERR_TYPE异常和报ERR_MSG
PADDLE_ENFORCE_LE cond_a <= cond_b 触发ERR_TYPE异常和报ERR_MSG
PADDLE_ENFORCE_NOT_NULL cond_a != nullptr 触发ERR_TYPE异常和报ERR_MSG

ERR_TYPE支持：

类型 含义
InvalidArgument 非法参数
NotFound 未找到
OutOfRange 越界
AlreadyExists 已存在
ResourceExhausted 资源超限
PreconditionNotMet 前置条件未满足
PermissionDenied 权限限制
ExecutionTimeout 超时
Unimplemented 未实现
Unavailable 不可用
Fatal Fatal错误
External 外部错误

ERR_MSG为C语言风格字符串，支持变长参数。

示例：

// 如果num_col_dims >= 2 && num_col_dims <= src.size()不为true则报InvalidArgument异常
// 和打印相关提示信息
PADDLE_ENFORCE_EQ(
      (num_col_dims >= 2 && num_col_dims <= src.size()),
      true,
      phi::errors::InvalidArgument("The num_col_dims should be inside [2, %d] "
                                   "in flatten_to_3d, but received %d.",
                                   src.size(),
                                   num_col_dims));

相关内容