人工智能基础数学类算子接口C代码参考定义示例.pdfVIP

GB/TXXXXX.1—XXXX

附录A

（资料性附录）

基础数学类算子接口C代码参考定义示例

A.1数据结构

#includestddef.h

#includestdint.h

/*ThetypecodeoptionsforDataType*/

typedefenum{

kChar0U,

kBool1U,

kInt2U,

kUInt3U,

kReal4U,

kComplex5U,

...

}TypeCode;

/*Thedatatypeoftheelementsinthetensor*/

typedefstruct{

/*Thetypecode*/

TypeCodecode;

/*Numberofbits,commonchoicesare1,2,4,8,16,32etc.*/

uint8_tsize;

}DataType;

/*ThedevicetypeoptionsforDevice*/

typedefenum{

kCPU0,

kGPU1,

…

}DeviceType;

/*Thedeviceofthetensorinmemory*/

typedefstruct{

/*Thedevicetype*/

DeviceTypetype;

/*Thedevice-idamongthisspecificdevicetype*/

intid;

}Device;

94

GB/TXXXXX.1—XXXX

/*Thelayoutformattypeofthetensorinmemory*/

typedefenum{

/*Densetensor*/

kDense,

/*CSRformatforsparsetensor*/

kCSR,

...

}LayoutType;

/*Thelayoutinformationofthetensorinmemory*/

typedefstruct{

/*Layouttype*/

LayoutTypetype;

/*Thelayoutdescriptionofthestoragetype*/

int64_t*min2maj;

}Layout;

/*TheShapeofthetensorinmemory*/

typedefstruct{

/*Numberofdimensions*/

int64_tndim;

/*Thedimensionsofthetensor*/

int64_t*dims;

/*Layoutinformationsoftensor*/

Layoutlayout;

}Shape;

/*Theimplementationofthestorageholdbythetensor*/

struct{

/*Thedatatypeoftheelementsinthestorage*/

DataTypedtype;

/*Thedevicewheretheelementsarestored*/

Devicedevice;

/*Thenumberoftheelementsholdbythestorage*/

int64_tsize;

/*Therawdatapointerpointingtothememoryspace*/

void*data;

}StorageImpl;