Dump日志接口
函数功能
记录AI CPU算子执行过程中的日志信息,便于功能调测。目前提供DEBUG/INFO/WARNING/ERROR级别的日志接口。
函数原型
CUST_KERNEL_DEBUG_LOG(ctx, fmt, ...)
CUST_KERNEL_INFO_LOG(ctx, fmt, ...)
CUST_KERNEL_WARNING_LOG(ctx, fmt, ...)
CUST_KERNEL_ERROR_LOG(ctx, fmt, ...)
参数说明
参数 |
输入/输出 |
说明 |
---|---|---|
ctx |
输入 |
CPUKernel的上下文信息,CpuKernelContext对象,算子Compute函数的入参。 |
fmt |
输入 |
格式控制字符串。 |
... |
输入 |
附加参数 ,根据不同的fmt 字符串,函数可能需要一系列的附加参数,每个参数包含了一个要被插入的值,替换了 fmt 参数中指定的每个%标签。 |
返回值说明
无。
约束说明
无。
调用示例
uint32_t UniqueCpuKernel::Compute(CpuKernelContext &ctx) { Tensor *param_tensor = ctx.Input(0); if (param_tensor == nullptr) { return 1; } auto param_shape = param_tensor->GetTensorShape(); if (param_shape == nullptr) { return 1; } int64_t p_size = 1; for (int i = 0; i < param_shape->GetDims(); ++i) { p_size *= param_shape->GetDimSize(i); } CUST_KERNEL_LOG_DEBUG(ctx, "Cust UniqueCpuKernel Compute, p_size is %ld.", p_size); ... }
父主题: AI CPU API