更新时间:2021-03-18 GMT+08:00
分享

编译依赖接口

如下接口只在算子编译过程中用到,用户在编写算子时不会直接调用。您可以在DDK包的安装目录下的“ddk/include/inc/tensor_engine/cce_aicpu_3rd_party.h”文件中查看接口的定义。

  • __aicpu__ void *aicpu_malloc(unsigned int size)

    该接口用于实现内存申请,入参size表示申请内存的大小,返回申请内存的指针。

  • __aicpu__ void aicpu_free(void * ptr)

    该接口用于实现内存释放,ptr为内存指针。

  • __aicpu__ double log(double x)

    该接口用于实现对x取log值,double类型。

  • __aicpu__ double exp(double x)

    该接口用于实现对x取exp值,double类型。

  • __aicpu__ double round(double x)

    该接口用于实现对x四舍五入,double类型。

  • __aicpu__ double floor(double x)

    该接口用于实现对x向下取整,double类型。

  • __aicpu__ double ceil(double x);

    该接口用于实现对x向上取整,double类型。

  • __aicpu__ double trunc(double x);

    该接口用于实现对x截断,double类型。

  • __aicpu__ double sqrt(double x);

    该接口用于实现对x开平方,double类型。

  • __aicpu__ float logf(float x)

    该接口用于实现对x取log值,double类型。

  • __aicpu__ float expf(float x)

    该接口用于实现对x取exp值,float类型。

  • __aicpu__ float roundf(float x)

    该接口用于实现对x四舍五入,float类型。

  • __aicpu__ float floorf(float x)

    该接口用于实现对x向下取整,float类型。

  • __aicpu__ float ceilf(float x);

    该接口用于实现对x向上取整,float类型。

  • __aicpu__ float truncf(float x);

    该接口用于实现对x截断,float类型。

  • __aicpu__ float sqrtf(float x);

    该接口用于实现对x开平方,float类型。

分享:

    相关文档

    相关产品