头文件列表

atb/types.h

基本数据结构和常量定义。

atb/svector.h

栈上的Vector定义。

atb/context.h

Context接口类，主要用于设置Operation上下文，如stream，tiling属性等。

atb/operation.h

算子类接口定义。

atb/infer_op_params.h

推理算子参数定义。

atb/train_op_params.h

训练算子参数定义。

atb/graph_op_builder.h

定义图算子的构建方法。

atb/utils.h

工具接口定义。

atb/atb_infer.h

仅包含对上述头文件的引用，引用该头文件即可提供对上述所有文件的引用。

atb/atb_comm.h

加速库通信域相关接口。