记录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);
...
}