文档
注册

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);
    ...
}
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词